Pytorch 一些函数用法

Posted 2021-01-02 cunyusup

PyTorch中view的用法:https://blog.csdn.net/york1996/article/details/81949843

max用法

import torch

d=torch.Tensor([[1,3],[2,4]])
print(d)

tensor([[1., 3.],
        [2., 4.]])

import torch

d=torch.Tensor([[1,3],[2,4]])
print(torch.max(d,0))
print(torch.max(d,1))

(tensor([2., 4.]), tensor([1, 1]))
(tensor([3., 4.]), tensor([1, 1]))

torch.max()返回两个结果，第一个是最大值，第二个是对应的索引值；第二个参数 0 代表按列取最大值并返回对应的行索引值，1 代表按行取最大值并返回对应的列索引值。