您的位置:首页 > 其它

PyTorch的concat也就是torch.cat实例

2017-12-05 11:14 507 查看
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]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: