项目分配张量流2.0类型错误:'tensorflow.python.framework.ops.Heartensor'对象不支持项分配人

2024-05-23 16:57:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用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”):

^{pr2}$

在Tensorflow 2.0中,有没有一种更简单/更有效的方法来完成EagerTensors的项分配?在


Tags: 项目方法importoutputtftensorflowasones