TensorFlow函数教程：tf.nn.weighted_moments

tf.nn.weighted_moments函数

tf.nn.weighted_moments(
    x,
    axes,
    frequency_weights,
    name=None,
    keep_dims=False
)

定义在：tensorflow/python/ops/nn_impl.py。

返回x的频率加权平均值和方差。

参数：

• x：一个Tensor。
• axes：值类型为int32的1-d张量；这些是计算均值和方差的轴。
• frequency_weights：可以用x广播的正权重的张量。
• name：用于范围操作的名称。
• keep_dims：生成与输入具有相同维度的moments。

返回：

两个Tensor：weighted_mean和weighted_variance。