怎么把一个(10,5)大小的tensor张量和一个(4,5)大小的tensor张量拼接为一个(14,5)大小的tensor张量
时间: 2024-05-13 10:14:03 浏览: 43
在keras 中获取张量 tensor 的维度大小实例
可以使用torch.cat函数来完成这个操作。代码如下:
```
import torch
tensor1 = torch.randn(10, 5)
tensor2 = torch.randn(4, 5)
concat_tensor = torch.cat((tensor1, tensor2), dim=0)
print(concat_tensor.shape) # (14, 5)
```
上述代码中,torch.cat函数接收一个元组作为参数,元组中的每个元素都代表一个需要拼接的tensor。dim参数表示在哪个维度上进行拼接,这里选择了0,表示在第一个维度上进行拼接。最后打印出concat_tensor的形状为(14, 5),即已经成功拼接。
阅读全文