Numpy与Torch中的size、shape、view等等那些事儿
最近脑子不好使,numpy和torch中的一些关于维度的函数及其功能总是搞混。为了记忆方便以及后续复习,暂且整理了些内容。
涉及到的函数/方法有:
np.size()、np.shape()、np.view()、np.reshape()、np.resize()、len(np.array)
tensor.size()、tensor.shape、tensor.view()、tensor.reshape()、tensor.resize()、tensor.nelement()、len(tensor)
其对应的主要功能整理如下,直接上图
相对应的例子如下:
以 [[1, 2, 3], [4, 5, 6]]为例。
关于array和tensor的维度变化上的函数不止这些,遇到又犯浑的时候再补充吧~
这是作为小白的个人笔记,内容上如有不正确之处,还请各位指正,互相交流。