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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。