I have a tensor A:
A = torch.rand(4,1,5,5)
with the size of:
torch.Size([4, 1, 5, 5])
and I have tensor B:
B = torch.rand(1,1,5,5)
With the size of:
torch.Size([1, 1, 5, 5])
Can I concatenate these two tensors in a way that i have a tensor A (or a new tensor) with size of:
torch.Size([4, 2, 5, 5])
In other words i want to concatenate in a way that add them to the dim = 1