row_sum = m.sum(dim=1) # shape (2,) row_sum_k = m.sum(dim=1, keepdim=True) # shape (2, 1) print(row_sum) # tensor([ 6., 15.]) print(row_sum_k) # tensor([[ 6.], # [15.]])