當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.raw_ops.ResourceScatterMin用法及代碼示例


使用min 操作減少對resource 引用的變量的稀疏更新。

用法

tf.raw_ops.ResourceScatterMin(
    resource, indices, updates, name=None
)

參數

  • resource Tensor 類型為 resource 。應該來自 Variable 節點。
  • indices 一個Tensor。必須是以下類型之一:int32 , int64ref 第一維的索引張量。
  • updates 一個Tensor。必須是以下類型之一:float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64。要添加到 ref 的更新值的張量。
  • name 操作的名稱(可選)。

返回

  • 創建的操作。

該操作計算

# Scalar indices
ref[indices, ...] = min(ref[indices, ...], updates[...])

# Vector indices (for each i)
ref[indices[i], ...] = min(ref[indices[i], ...], updates[i, ...])

# High rank indices (for each i, ..., j)
ref[indices[i, ..., j], ...] = min(ref[indices[i, ..., j], ...], updates[i, ..., j, ...])

正確處理重複條目:如果多個 indices 引用相同的位置,則它們的貢獻被合並。

需要 updates.shape = indices.shape + ref.shape[1:]updates.shape = []

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.raw_ops.ResourceScatterMin。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。