如何相等大小为[64、3、128、128]和[64、64、127、127]的两个张量的维

Question

我正在尝试在pytorch中添加两个维度为[64, 64, 127, 127]和[64, 3, 127, 127]的张量，但是我需要在第二个张量中添加零，以便将它们的大小等于[64, 64, 127, 127]。 / p>

我尝试通过切片添加零，但是它不可靠

Answer 1

第一个很容易可视化：

import torch
zeros = torch.zeros(2, 2, 2, 10)
ones = torch.ones(2, 2, 2, 5)

zeros[:, :, :, :5] = ones
print(ones)
print(zeros)

第二个是您需要的（零和一个仅用于我的测试）

ones = torch.ones(64, 3, 127, 127)
zeros = torch.zeros(64, 61, 127, 127)
zeros[:, :3, :, :] = ones

使用cat()函数的简单操作如下：

torch.cat((ones, zeros), 1)