烨然的博客

1 torch.cat torch.cat((A, B), dim) 将两个tensor在指定维度进行拼接 A = torch.zeros(2,3) B = torch.zeros(2,3) C = torch.cat((A,B), 0) ## shape [4,3] D = torch.cat((A,B), 1) ## shape [2,6] 2 torch....