Python tf.math.reduce_variance用法及代碼示例

計算張量維度上元素的方差。

用法

tf.math.reduce_variance(
    input_tensor, axis=None, keepdims=False, name=None
)

沿 axis 中給定的尺寸減少 input_tensor 。除非 keepdims 為真，否則對於 axis 中的每個條目，張量的秩都會減少 1，這必須是唯一的。如果 keepdims 為真，則保留縮減後的維度，長度為 1。

如果axis 為None，則所有維度都會減少，並返回具有單個元素的張量。

例如：

x = tf.constant([[1., 2.], [3., 4.]])
tf.math.reduce_variance(x)
<tf.Tensor:shape=(), dtype=float32, numpy=1.25>
tf.math.reduce_variance(x, 0)
<tf.Tensor:shape=(2,), dtype=float32, numpy=array([1., 1.], ...)>
tf.math.reduce_variance(x, 1)
<tf.Tensor:shape=(2,), dtype=float32, numpy=array([0.25, 0.25], ...)>

numpy 兼容性

相當於 np.var

請注意np.var 有一個dtype 參數，可用於指定輸出類型。默認情況下這是 dtype=float64 。另一方麵，tf.math.reduce_variance 具有來自 input_tensor 的激進類型推斷。