Tensorflow–取tensorf指定列的操作方式

我就废话不多说了,大家还是直接看代码吧~ In [1]: import os In [2]: os.environ[CUDA_VISIBLE_DEVICES] = 0 In [3]: import tensorflow as tf In [4]:sess =tf.Session() In [5]: input = tf.constant([[[1,2,3],[4,5,6],[7,8,9]],[[10,11,12],[13,14,15],[1 ...: 6,17,18]]]) In [6]: input.get_shape() Out[6]: TensorShape([Dimensi