网站介绍：文章浏览阅读1.1k次。目录torch常用API1.torch.max(x,dim=1)1.1定义1.2exampletorch常用API1.torch.max(x,dim=1)1.1定义 input输入的是一个tensor dim是max函数索引的维度0/1，0是每列的最大值，1是每行的最大值 返回的是两个值：一个是每一行最大值的tensor组，另一个是最大值所在的位置 max_col_value = torch.max(x,dim=0)[0] # 每一..._torch.max(x, dim=1, keepdim=true)[0]
- 链接地址：https://blog.csdn.net/weixin_41297324/article/details/108144989
- 链接标题：Pytorch(二)：torch常用API应用及详解_torch.max(x, dim=1, keepdim=true)[0]-CSDN博客
- 所属网站：blog.csdn.net
- 被收藏次数：432
- 网站标签：torch.max(x dim=1 keepdim=true)[0]
