一、builtin_function_or_method object has no attribute ‘size’

最近在学注意力机制,再编写Senet的时候,这个bug最开始是长这样子的:
在这里插入图片描述
它意思就是,往里面传的的参数x有问题(x是有4个维度)。结果我找了半天没找到,最后发现是input写错了:
在这里插入图片描述
iuput 牛皮

二、获取张量的四个维度 torch里的size()函数

size()可以获得Tensor的四个维度,具体操作如下:
(b,c,h,w分别batch数,通道数,高,宽)

import torch
zl=torch.ones(3,20,28,28)
b,c,h,w=zl.size()
print('张量的batch维度是{}'.format(b))
print('张量的通道维度是{}'.format(c))

在这里插入图片描述
当然也可以size(),括号里面写上具体的数,返回指定的维度数
比如size(0),返回batch维度数,size(1)返回通道维度数

三、改变tensor形状的函数view()

首先,我们生成一个通道数为2,高为3,宽为4的张量:(共有234个元素)

import torch
zl=torch.ones(2,3,4) #这个张量共有2*3*4=24个元素
print(zl)

在这里插入图片描述
先讲讲最基本的方法吧,把这24个元素编排成4*6这样的

zl1=zl.view(4,6)
print(zl1)

在这里插入图片描述
当然也可以1* 4* 6这样,有点不一样

zl2=zl.view(1,4,6)
print(zl2)

在这里插入图片描述
在重新编排形状的时候,一定要记得元素个数一定要一致,否则就乱了,比如下面这种:

zl2=zl.view(2,60)
print(zl2)

在这里插入图片描述
当然,还有一些用的多的方法:

zl4=zl.view(-1)
print(zl4)

在这里插入图片描述

zl5=zl.view(-1,1)
print(zl5)

在这里插入图片描述
(图片没有截取完全,有24排)

zl6=zl.view(zl.size(0),-1)
print(zl6)

在这里插入图片描述

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐