对于一些自定义代码,我需要运行一个For循环在Tensorflow 2中动态创建一个变量(启用了急切执行模式)。import numpy as npimport timeit
data = np.random.randn(N)
def,在for循环中,我只需要更新变量的一小部分。在每次迭代中,要更新的切片是不同的。用于更新的数据在每次迭代中都是不同的(在我的自定义代码中,它是依赖于变量切片的一
使用tensorflow完成此操作的最佳方式是什么?我的尝试:
result = tf.empty([B, N, N, 5]) for b in range(B): for i in range(4) result[b, :, :, i] =tf.mul(T[b, :, :, i], T[b, :, :, 5]) result[b, :, :, 4] = T[b, :, :, 4]