TensorFlow函数:tf.sparse_segment_mean

tf.sparse_segment_mean 函数

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

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

请参阅指南:数学>分段

tf.sparse_segment_mean函数计算沿张量稀疏段的平均值.

与 SegmentMean 相同,但 segment_ids 可以有低于 data 第一维的秩,选择维度 0 的子集,由 indices 指定.segment_ids 允许缺少 ID,在这种情况下,输出将在这些索引中为零.在这些情况下,num_segments 用于确定输出的大小.

函数参数:

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

函数返回值:

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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部