torch.sum()

1. target tensor

2. dimension index

• keepdim - If this is True, the output tensor is of the same size as input except in the dimension(s) dim where it is of size 1. Else, dim is squeezed (see torch.squeeze()), resulting in the output tensor having 1 (or more) fewer dimension(s).

• dtype - Specifies the desired data type of returned tensor. If specified, the input tensor is casted to dtype before the operation is performed.

make average for each row