pytorch学习笔记——常用函数

torch.softmax()
torch.reduce_max()
torch.pow()
torch.subtrack()
torch.gather()
torch.log()

你可能感兴趣的:(pytorch学习笔记,pytorch)