torch.max的用法

torch.max如何使用?

看下面的例子

torch.max的用法_第1张图片
需要注意的是,torch.max返回“最大值和”对应的位置,不要以为仅仅返回最大值就完事了。

你可能感兴趣的:(python3,Linux,机器学习,深度学习)