2020
10-08
10-08
pytorch 常用函数 max ,eq说明
max找出tensor的行或者列最大的值:找出每行的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,1))输出:(tensor([1.,2.,3.]),tensor([0,0,0]))找出每列的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,0))输出结果:(tensor([3.]),tensor([2]))Tensor比较eq相等:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])targe...
继续阅读 >