一、定义

  • torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。

二、例子

三、注意事项

使用torch.cat((A,B),dim)时,除拼接维数dim数值可不同外其余维数数值需相同,方能对齐。

Logo

一站式 AI 云服务平台

更多推荐