torch.bmm()强制规定维度和大小相同
torch.matmul()没有强制规定维度和大小,可以用利用广播机制进行不同维度的相乘操作
当进行操作的两个tensor都是3D时,两者等同。
类的call方法调用以及dataloader中的collate_fn
class aaa(object):
def __init__(self,bbb,ccc):
...
def __call__(self,batch):
...
__call__相当于重载了括号运算符,加入call方法后,aaa类直接可以像函数一样调用,如果括号不加参数,就等同于一个函数。
如_aaa = aaa(bbb=b,ccc=c)
Dataloader中的collate_fn参数可以直接引入_aaa,则get_item即dataloader每次迭代返回的batch为按collate_fn的即aaa中call方法定义的规则处理过的样本组合的一个list。