paddle框架下两个1维的tensor,组成一个2维的tensor,请代码展示
时间: 2023-06-01 18:07:53 浏览: 309
采用paddle框架自定制一维卷积神经网络对西储大学轴承数据集进行故障识别(复现于基于卷积神经网络的轴承故障诊断算法研究的论文)
可以使用PaddlePaddle框架中的concat函数将两个1维的tensor组成一个2维的tensor,具体代码如下:
```python
import paddle
x = paddle.to_tensor([1, 2, 3]) # 创建第一个1维的tensor
y = paddle.to_tensor([4, 5, 6]) # 创建第二个1维的tensor
# 使用concat函数将两个1维的tensor组成一个2维的tensor
z = paddle.concat([x, y], axis=0).unsqueeze(0)
print(z)
```
输出结果为:
```
Tensor(shape=[1, 2, 3], dtype=int64, place=CUDAPlace(0), stop_gradient=True,
[[[1, 2, 3],
[4, 5, 6]]])
```
其中axis=0表示按照第0个维度(也就是竖直方向)连接两个tensor,unsqueeze(0)函数将2维的tensor转换为3维的tensor。
阅读全文