【Tensorflow】tf.stack() tf.unstack()

一。 import tensorflow as tf      a = tf.constant([1,2,3])   b = tf.constant([4,5,6])      c = tf.stack([a,b],axis=0)   d = tf.stack([a,b],axis=1)      e = tf.unstack(d,axis=0)   f = tf.unstack(d,axis=1
相关文章
相关标签/搜索