pytorch cat
时间: 2023-10-18 19:25:23 浏览: 124
PyTorch的torch.cat用法
5星 · 资源好评率100%
PyTorch的`cat`函数是用于将多个张量按指定维度进行拼接的函数。它的使用方法如下:
```python
import torch
# 创建两个张量
x1 = torch.randn(3, 4) # 3行4列的张量
x2 = torch.randn(3, 4)
# 拼接两个张量
result = torch.cat((x1, x2), dim=0) # 在行维度上进行拼接
print(result)
```
这里,我们创建了两个形状相同的张量`x1`和`x2`,然后使用`cat`函数将它们在行维度上进行拼接。结果是一个新的张量,其行数为两个输入张量行数之和,列数保持不变。
如果要在不同的维度上进行拼接,可以通过修改`dim`参数来实现。例如,`dim=1`表示在列维度上进行拼接。
希望这个回答能够满足你的需求!如果还有其他问题,请随时提问。
阅读全文