torch.max(input)
返回input矩阵的最大值
torch.max(input,dim)
返回类型为元组(namedtuple),第一个元素为值,第二个元素为对应索引
dim = 0 按照列取最大值和索引并返回
dim = 1 按照行取最大值和索引并返回
squeeze()
squeeze(input, dim=None),如果不给定dim,则把input的所有size为1的维度给移除;如果给定dim,则只移除给定的且size为1的维度。
torch.max(input)
返回input矩阵的最大值
torch.max(input,dim)
返回类型为元组(namedtuple),第一个元素为值,第二个元素为对应索引
dim = 0 按照列取最大值和索引并返回
dim = 1 按照行取最大值和索引并返回
squeeze(input, dim=None),如果不给定dim,则把input的所有size为1的维度给移除;如果给定dim,则只移除给定的且size为1的维度。