默认情况下,调用求和函数会沿所有的轴降低张量的维度,使它变为一个标量。 可以指定张量沿哪一个轴来通过求和降低维度。
以矩阵为例,通过求所有行元素的和来降维(轴0),即将相同列位置对应的所有行位置元素相加,列数保持不变,可以在调用函数时指定axis=0。
axis=1,是通过求所有列元素的和来降维(轴1),即将相同行位置对应的所有列位置元素相加,行数保持不变。
import torch
T = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])
print("axis = 0: ",torch.sum(T,axis = 0))
print("axis = 1: ", torch.sum(T,axis = 1))
输出:
axis = 0: tensor([12, 15, 18])
axis = 1: tensor([ 6, 15, 24])