当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.raw_ops.TensorScatterMax用法及代码示例


将稀疏更新应用于获取元素最大值的张量。

用法

tf.raw_ops.TensorScatterMax(
    tensor, indices, updates, name=None
)

参数

  • tensor 一个Tensor。要更新的张量。
  • indices 一个Tensor。必须是以下类型之一:int32 , int64。索引张量。
  • updates 一个Tensor。必须与 tensor 具有相同的类型。更新分散到输出中。
  • name 操作的名称(可选)。

返回

  • 一个Tensor。具有与 tensor 相同的类型。

返回从 tensor 复制的新张量,其值是张量和根据索引更新之间的元素最大值。

tensor = [0, 0, 0, 0, 0, 0, 0, 0]
indices = [[1], [4], [5]]
updates = [1, -1, 1]
tf.tensor_scatter_nd_max(tensor, indices, updates).numpy()
array([0, 1, 0, 0, 0, 1, 0, 0], dtype=int32)

有关详细信息,请参阅tf.tensor_scatter_nd_update

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.TensorScatterMax。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。