如何更改火炬张量以与另一个张量连接

【问题标题】：How do I change a torch tensor to concat with another tensor如何更改火炬张量以与另一个张量连接
【发布时间】：2020-04-16 22:48:15
【问题描述】：

我正在尝试将数值数据的张量与 resnet-50 模型的输出张量连接起来。该模型的输出是张量形状torch.Size([10,1000])，数值数据是张量形状torch.Size([10, 110528,8])，其中10 是批量大小，110528 是数据框意义上的观察数，8 是数量列数（在数据框意义上）。我需要将数值张量重塑为torch.Size([10,8])，以便正确连接。

我将如何重塑张量？

【问题讨论】：

可能在 SE 交叉验证上问这个。

标签： pytorch

【解决方案1】：

起始张量。

a = torch.randn(10, 1000)
b = torch.randn(10, 110528, 8)

允许连接的新张量。

c = torch.zeros(10,1000,7)

检查形状。

a[:,:,None].shape, c.shape

(torch.Size([10, 1000, 1]), torch.Size([10, 1000, 7]))

更改张量 a 以允许连接。

a = torch.cat([a[:,:,None],c], dim=2)

在维度 1 中连接。

torch.cat([a,b], dim=1).shape

torch.Size([10, 111528, 8])

【讨论】：