网站怎么做分站,小型网站设计及建设, 天堂中文在线,广州仿网站官方说明torch.cat 引用自#xff1a;Pytorch中的torch.cat()函数 torch.cat(tensors, dim0, *, outNone) → Tensor
# 连接给定维数的给定序列的序列张量。所有张量要么具有相同的形状(除了连接维度)#xff0c;要么为空。示例 输入#xff1a; import torch
a torch.Tens…官方说明torch.cat 引用自Pytorch中的torch.cat()函数 torch.cat(tensors, dim0, *, outNone) → Tensor
# 连接给定维数的给定序列的序列张量。所有张量要么具有相同的形状(除了连接维度)要么为空。示例 输入 import torch
a torch.Tensor(2,3) # (2行3列)
b torch.Tensor(2,3)
print (a)
print (b)输出 tensor([[8.9082e-39, 1.0194e-38, 9.1837e-39],[8.4490e-39, 9.6429e-39, 8.4490e-39]])
tensor([[-2.0541e-05, 5.0727e-43, -2.0541e-05],[ 5.0727e-43, -2.1039e-05, 5.0727e-43]])输入 print(torch.cat([a,b], dim 0))
# 1. torch.cat((x,y),dim0) :张量 X,Y按照列堆起来输出 tensor([[ 8.9082e-39, 1.0194e-38, 9.1837e-39],[ 8.4490e-39, 9.6429e-39, 8.4490e-39],[-2.0541e-05, 5.0727e-43, -2.0541e-05],[ 5.0727e-43, -2.1039e-05, 5.0727e-43]])输入 print(torch.cat([a,b], dim-1))
# 2. torch.cat((x,y),dim1) :张量 X,Y按照行并排起来输出 tensor([[ 8.9082e-39, 1.0194e-38, 9.1837e-39, -2.0541e-05, 5.0727e-43,-2.0541e-05],[ 8.4490e-39, 9.6429e-39, 8.4490e-39, 5.0727e-43, -2.1039e-05,5.0727e-43]])总结
torch.cat((x,y),dim0) :张量 X,Y按照列堆起来 torch.cat((x,y),dim1) :张量 X,Y按照行并排起来
