mean()函数的参数:dim=0,按行求平均值,返回的形状是(1,列数);dim=1,按列求平均值,返回的形状是(行数,1),默认不设置dim的时候,返回的是所有元素的平均值。
需要注意的是 mean()函数只能在float格式的数据上处理
如果不是,要不就x = x.float()
或者定义的时候 dtype=torch.float
mean()函数的参数:dim=0,按行求平均值,返回的形状是(1,列数);dim=1,按列求平均值,返回的形状是(行数,1),默认不设置dim的时候,返回的是所有元素的平均值。
需要注意的是 mean()函数只能在float格式的数据上处理
如果不是,要不就x = x.float()
或者定义的时候 dtype=torch.float