我正在和tensorflow一起研究一个算法。以下是想要的代码的NumPy版本: x = [1,2,3,4,5,6,7,8,9,10]y = [0,0,0,0,0,0,0,0,0,0]
if i in sets:
y[i] = x[i] 它会得到结果: y = [0,2,0,0,0,6,0,8,0,0] 如何在tensorflow中实现此功能?有没有办法使用相同的逻辑在tensorf
假设张量T的形状为[B, N, N, 6],我想为每个b in range(B)逐个元素地将矩阵[b, N, N, 0:3]乘以[b, N, N, 5]。请注意,不应更改[N, N, 4]。使用tensorflow完成此操作的最佳方式是什么?我的尝试:
result = tf.empty([B, N, N, 5]) for b in range(B): for i in range(4) result[b, :, :, i] =