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

根據 indices 從 params 收集切片。

用法

tf.raw_ops.Gather(
    params, indices, validate_indices=True, name=None
)

indices 必須是任意維度的整數張量(通常為 0-D 或 1-D)。生成形狀為 indices.shape + params.shape[1:] 的輸出張量，其中：

# Scalar indices
    output[:, ...,:] = params[indices,:, ...:]

    # Vector indices
    output[i,:, ...,:] = params[indices[i],:, ...:]

    # Higher rank indices
    output[i, ..., j,:, ...:] = params[indices[i, ..., j],:, ...,:]

如果indices 是一個排列並且len(indices) == params.shape[0] 那麽這個操作將相應地排列params。

validate_indices：已棄用。如果將此操作分配給 CPU，則始終驗證 indices 中的值在範圍內。如果分配給 GPU，越界索引會導致安全但未指定的行為，其中可能包括引發錯誤。