pytorch中的cat、stack、tranpose、permute、unsqeeze

pytorch中提供了对tensor经常使用的变换操做。web cat 链接 对数据沿着某一维度进行拼接。cat后数据的总维数不变。 好比下面代码对两个2维tensor(分别为2*3,1*3)进行拼接,拼接完后变为3*3仍是2维的tensor。 代码以下:svg import torch torch.manual_seed(1) x = torch.randn(2,3) y = torch.ran
相关文章
相关标签/搜索