torch.
max
(input) → Tensor
返回输入tensor中所有元素的最大值
torch.
max
(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)
按维度dim 返回最大值
torch.max(a, dim = 1) 返回每一行中最大值的那个元素values,且返回其索引indices(返回最大元素在这一行的列索引)
torch.
max
(input) → Tensor
返回输入tensor中所有元素的最大值
torch.
max
(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)
torch.max(a, dim = 1) 返回每一行中最大值的那个元素values,且返回其索引indices(返回最大元素在这一行的列索引)