Tensorflow切片

时间:2017-10-04 11:55:54

标签: python indexing tensorflow

我尝试使用tf.Tensor.getitem函数切割张量,如下所示:

indices = [0, 5]
data[:,:,indices]

但是我收到以下错误:

  

TypeError:只能将列表(不是“int”)连接到列表。

我做错了吗?

1 个答案:

答案 0 :(得分:1)

您可以尝试以下操作:

data[:, :, 0:6]