经过一段时间的Pytorch使用,发现提供的Tensor类型中大部分操作都很好理解,比如各种数学操作(abs,add,asin,mul,div)和它们组成的复合操作(addcdiv,addcmul)等。
下面将对于写代码时,我遇到的一些常用的“特殊”操作,总结mark下:
clamp 截断
用来“截断”tensor,使每个元素都保持在min~max范围内
torch.clamp(input,min,max,out=None)
cat 合并
沿着第dim维度将seq中的tensor合并成一个tensor
torch.cat(seq,dim=0,out=None)
chunk 分块
按照第dim维,将tensor分成chunks块(如果不能整除的话,则最后一块会略小,比如该维度为3则分成2和1)
torch.chunk(tensor,chunks