torch.nn.argmax()
时间: 2024-05-18 08:16:28 浏览: 158
torch.nn.argmax()函数是不存在的。在PyTorch中,没有torch.nn模块中的argmax()函数。然而,可以使用torch.argmax()函数来实现相同的功能。该函数接受张量作为输入,并返回指定维度上的最大值的索引。您可以通过指定dim参数来选择要减少的维度,从而计算该维度上的最大值的索引。例如,如果您有一个二维张量x,您可以使用torch.argmax(x, dim=0)来计算dim=0维度上的最大值的索引,使用torch.argmax(x, dim=1)来计算dim=1维度上的最大值的索引。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *3* [torch.argmax()函数](https://blog.csdn.net/qq_40507857/article/details/123357918)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
- *2* [perturbations-differential-pytorch:在Pytorch中具有扰动的可微分优化器](https://download.csdn.net/download/weixin_42172204/16301724)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]
阅读全文