Function WeightedAvg

Computes weighted average of values from a tensor. Args: values: a tensor of values. weights: a tensor of weights. sum_reduction_fn: called to reduce the values and weights. axis: the axis to reduce over; defaults to all (i.e. to a scalar). name: name of metric. Returns:

(
    values: tf.Tensor,
    weights: tf.Tensor,
    sum_reduction_fn: ReductionProtocol = tf.reduce_sum,
    axis: Optional[int] = None,
    name: str = '',
)