标签：02 03 01 tensor argmax pred but range expected

在pytorch训练模型的时候，出现：

代码：

            pred = model(X)
            print(pred.argmax(1))

打印pred为：

tensor([ 17.0364,  28.3838, -27.5744,   8.5920])

因为只有一维，所以需要改为：

print(pred.argmax(0))

这样就没有问题了

当我们一次使用多个输入数据时，可能tensor就是二维的，这个时候才可以用pred.argmax(1)得到最大值的索引。

tensor([[ 1.1916e-01, -1.7842e-01,  2.4500e-01, -1.1631e-01,  4.5129e-01,
         -2.1620e-01,  2.5249e-01, -3.0434e-01,  1.0978e-01,  4.3598e-02],
        [ 4.3276e-02, -1.3183e-02,  4.3428e-02, -4.3271e-03,  9.0060e-02,
         -2.3623e-02,  6.0345e-03, -2.6779e-02,  6.1037e-02,  2.1716e-02],
        [ 8.4877e-02,  7.0410e-03,  7.7200e-02,  1.4489e-02,  1.6732e-01,
         -6.9628e-02,  5.2289e-02, -1.0901e-01,  2.8091e-02, -5.3942e-03]])

标签：02,03,01,tensor,argmax,pred,but,range,expected
来源： https://blog.csdn.net/tiao_god/article/details/120623404

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。