愉快发箍

文章
9
资源
0
加入时间
3年0月27天

2021.08.22学习内容torch.cat()和torch.stack()函数

torch.cat()函数将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。def cat(tensors: List[torch.Tensor], dim: int = 0)C = torch.cat( (A,B),0 ) #按维数0拼接(竖着拼)(dim=0) C = torch.cat( (A,B),1 ) #按维数1拼接(横着拼)(dim=1)torch.stack()函数这两者的区别在于torch.cat沿着给定的维度拼接,而torc