tf.scatter_update函數(shù)：將稀疏更新應用于變量引用

tf.scatter_update 函數(shù)

scatter_update(
    ref,
    indices,
    updates,
    use_locking=True,
    name=None
)

請參閱指南：變量>稀疏變量更新

將稀疏更新應用于變量引用.

此操作計算如下：

# Scalar indices
ref[indices, ...] = updates[...]

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

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

此操作在更新完成后輸出 ref.這樣可以更容易地鏈接到需要使用重置值的操作.

如果 ref 中的值要更新多次,由于indices中有重復的條目,則每個值的更新發(fā)生的順序是未定義的.

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

函數(shù)參數(shù)

• ref：一個可變的Tensor；應該來自一個Variable節(jié)點.
• indices：一個Tensor；必須是以下類型之一：int32,int64；進入ref的第一維度的一個索引的張量.
• updates：一個Tensor.必須與ref具有相同的類型.存儲在ref中的更新值的張量.
• use_locking：可選bool,默認為True；如果為True,則賦值將受鎖定的保護；否則行為是不確定的,但可能表現(xiàn)出較少的爭用.
• name：操作的名稱(可選).

函數(shù)返回值

和ref一樣；作為在更新完成后想要使用更新值的操作的便利返回.