如何将此np.add.at语句转换为TensorFlow?

时间:2018-03-05 23:35:55

标签: python numpy tensorflow

np.add.at(img, (slice(None), z, y, x), cols_reshaped)

img是4d

cols_reshaped是2d

zyx都是2d

1 个答案:

答案 0 :(得分:0)

截至2018年3月,在切片操作和变异方面,Tensorflow比numpy更受限制。 Tensorflow张量是不可变的(有助于区分),因此如果要改变变量,则需要使用变量。我认为你可以通过链接tf的strided_slice和strided_slice_assign功能来模拟这个。