Python tf.tensor_scatter_nd_max用法及代码示例

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

用法

tf.tensor_scatter_nd_max(
    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.tensor_scatter_nd_max。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。