每天五分钟深度学习pytorch：一文帮你搞懂维度操作

本文重点

张量的操作有很多，比如切分合并，在某一维度取值，此时张量的维度信息让人很难快速确定，本文对其进行简单的总结。

例子

kthvalue是求指定维度的第几小值，这个不重要，现在a的shape=4，5也就是4行5列，在dim=1上操作，那么就是从列上操作，也就是每一列上取第三小的值，那么一共有5列，那么就会有5个值。

可是程度一种取出了四个值，那么说明上面的思维是错误的，个人建议当遇到这种问题的时候，不要将其展开想，记住一句话，在哪个维度上操作，结果就是哪个维度消失，比如a的shape=4，5在dim=1上操作，也就是5消失，那么结果的shape就是4.

或者这样来思考这个问题：

shape=4，5