建立一个4×2的矩阵c, c.shape[1] 为第一维的长度,c.shape[0] 为第二维的长度。但是用维度来解释总是觉得很绕,尤其是在我看到shape[3]的时候
所以,我索性这么理解,在pytorch中的图片一般是以(Batch_size,通道数,图片高度,图片宽度)或者(Batch_size,通道数,图片宽度,图片高度),那么shape[0]可以理解为Batch_size,shape[1]可以理解为通道数,以此类推
关于pytorch中的shape[]维度的问题
最新推荐文章于 2024-07-02 23:17:18 发布