tf.math.reduce_sum

Computes the sum of elements across dimensions of a tensor.

Used in the notebooks