import torch
a = torch.ones([1,2])
print(a)
# tensor([[1., 1.]])
b = torch.ones([1,2])
print(b)
# tensor([[1., 1.]])
print(torch.cat([a,b],1))
# tensor([[1., 1., 1., 1.]])
torch.cat就是将a,b 按行放在一起,如果第二个参数是0,则按列放在一起。
import torch
a = torch.ones([1,2])
print(a)
# tensor([[1., 1.]])
b = torch.ones([1,2])
print(b)
# tensor([[1., 1.]])
print(torch.cat([a,b],1))
# tensor([[1., 1., 1., 1.]])
torch.cat就是将a,b 按行放在一起,如果第二个参数是0,则按列放在一起。