argmax
torch.argmax()函数-截至2023年8月28日
argmax函数参数dim=0表示从列获取最大值索引，dim=1从行获取最大值索引，dim=-1从最后一个维度获取最大值索引[^argmax参考]。 举例 ```python import torch d = torch.tensor([[9,7,6], [4,8,2], [5,10,0]]) pr ......
【d2l】【常见函数】【3】 tensor.argmax( )
**从张量的特定维度上取出最大的元素值所对应的索引**  ![image](https://img202 ......
argmax经过sigmoid和不经过sigmoid区别
title: argmax经过sigmoid和不经过sigmoid区别 date: 2023-04-05 21:49:24 tags: - 深度学习 - argmax - sigmoid categories: - 技术 起因 今天和同组讨论了一下网络输出时，在torch.argmax之前经过tor ......