使用Tensorflow中的索引切片张量

Question

基本上我有一个2d数组,我想做这个很好的numpy式的东西

noise_spec[:rows,:cols]

在Tensorflow.行和列只是两个整数.

Answer 1

实际上,TensorFlow现在可以更好地支持切片,因此您可以使用与NumPy完全相同的语法:

result = noise_spec[:rows, :cols]

Answer 2

发现了，是

tf.slice(noise_spec, [0,0],[rows, cols])