我正在使用Tensorflow 2.0,并尝试在我的张量中更新切片。在
使用Pythorch中的纯项目分配,它不起作用。在
import tensorflow as tf
tensor = tf.ones((10, 192, 85))
tensor[:, :, 0] = tf.math.sigmoid([:, :, 0])
>>> Output
TypeError: 'tensorflow.python.framework.ops.EagerTensor' object does not support item assignment
我知道可以使用tf.tensor_scatter_nd_update
,但它太冗长了,效率不高,因为我必须分别派生要更新的索引。因此,我不确定这是否是在急切张量中进行项分配的最佳方法(我需要下面的代码块来实现上面更简单的“PyTorch 2-liner style”):
在Tensorflow 2.0中,有没有一种更简单/更有效的方法来完成EagerTensor
s的项分配?在
目前没有回答
相关问题 更多 >
编程相关推荐