【学习笔记】关于维度,unsqueeze,squeeze

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zzzpy/article/details/88076939

squeeze和unsqueeze可以参考:https://blog.csdn.net/xiexu911/article/details/80820028

图片来源:https://blog.csdn.net/flysky_jay/article/details/81607289

维度是(2,3)

维度是(2,1,3)

我一般这样理解维度,第一维是去掉第一个【】,剩下的元素个数(可能是子【】),以此类推。

猜你喜欢

转载自blog.csdn.net/zzzpy/article/details/88076939