神奇的max函数,你品,你细品
import torch
a = torch.randn(1,10)
print(a)
print("-------------------")
out = torch.max(a,1)
print(out)
print("-------------------")
out = torch.max(a,1)[1]
print(out)
print("-------------------")
out = torch.max(a,1)[1<
神奇的max函数,你品,你细品
import torch
a = torch.randn(1,10)
print(a)
print("-------------------")
out = torch.max(a,1)
print(out)
print("-------------------")
out = torch.max(a,1)[1]
print(out)
print("-------------------")
out = torch.max(a,1)[1<