TensorFlow函数:tf.sparse_segment_sqrt_n

由 Carrie 创建, 最后一次修改 2018-03-08

tf.sparse_segment_sqrt_n 函数

sparse_segment_sqrt_n(
    data,
    indices,
    segment_ids,
    name=None,
    num_segments=None
)

定义在:tensorflow/python/ops/math_ops.py.

请参阅指南:数学>分段

计算由 sqrt (N) 除以的张量的稀疏段的总和.
N 是要减少的线段的大小.

函数参数:

  • data:有数据的 Tensor,将在输出中组合.
  • indices:一个 1-D 的 Tensor,它带有索引 data.与 segment_ids 有相同的秩.
  • segment_ids:一个 1-D 的 Tensor,它带有索引的输出 Tensor,值应该排序并且可以重复.
  • name:操作的名称(可选).
  • num_segments:一个可选的 int32 标量.指示输出 Tensor 的大小.

函数返回值:

该函数返回作为数据的形状的一个 tensor,除了大小为 k 的维度 0 之外,它通过 num_segments 指定的段的数量或在 segments_ids 中推断最后一个元素.

以上内容是否对您有帮助:
二维码
建议反馈
二维码