torch.softmax()
torch.reduce_max()
torch.pow()
torch.subtrack()=torch.sub()
torch.div()=torch.divide()
torch.gather()
torch.log()
pytorch常用计算

更多推荐