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，越界索引会导致安全但未指定的行为，其中可能包括引发错误。