【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
相關文章
相關標籤/搜索