PyTorch的concat也就是torch.cat实例

import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.cat([a,b],1)
 1  1  1  1
[torch.FloatTensor of size 1x4]
相关推荐
©️2020 CSDN 皮肤主题: 创作都市 设计师:CSDN官方博客 返回首页