【pytorch学习】torch.cat
因为图像识别中网络的Tensor一般为N * C * H * W,所以我们的例子也是用4维数据
维度是从零开始的,即生成一个a = torch.randn(1, 3, 3, 4)
首先我们先从维度1开始拼接数据,即Tensor的维度C(channel)
import torch
a = torch.randn(1, 3, 3, 4) # C 为3
b = torch.ra...
原创
2019-04-11 10:29:29 ·
699 阅读 ·
0 评论