- 博客(7)
- 收藏
- 关注
原创 torch.argmax()函数
dim表示不同维度。特别的在dim=0表示二维矩阵中的列,dim=1在二维矩阵中的行。返回指定维度最大值的序号。
2023-07-16 15:26:02
168
转载 Pytorch中Tensor的类型转换
https://www.cnblogs.com/sbj123456789/p/10839020.html
2023-03-11 10:05:04
1145
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅
1