TensorFlow函數(shù)教程：tf.nn.weighted_moments

tf.nn.weighted_moments函數(shù)

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

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

返回x的頻率加權(quán)平均值和方差。

參數(shù)：

• x：一個Tensor。
• axes：值類型為int32的1-d張量；這些是計算均值和方差的軸。
• frequency_weights：可以用x廣播的正權(quán)重的張量。
• name：用于范圍操作的名稱。
• keep_dims：生成與輸入具有相同維度的moments。

返回：

兩個Tensor：weighted_mean和weighted_variance。