paddle.unsqueeze
时间: 2023-09-25 09:07:21 浏览: 89
Paddle.js 是百度 PaddlePaddle 的 web 方向子项目,是一个运行在浏览器中的开源深度学习框架
paddle.unsqueeze是一个PaddlePaddle的函数,用于在Tensor的指定位置增加一个新的维度。具体地说,unsqueeze函数可以用来扩展Tensor的维度,使其在指定位置增加一个大小为1的维度。该函数的使用方法如下:
```python
paddle.unsqueeze(input, axis)
```
其中,input是输入的Tensor,axis是要在哪个位置增加新的维度。axis的取值范围是从0开始,表示插入的维度的索引位置。
举个例子,假设我们有一个形状为(3, 4)的Tensor,我们可以使用unsqueeze函数在第二个维度上增加一个新的维度,代码如下:
```python
import paddle
x = paddle.randn([3, 4])
y = paddle.unsqueeze(x, axis=1)
```
上述代码中,x是一个形状为(3, 4)的Tensor,使用unsqueeze函数在axis=1的位置上增加了一个新的维度,结果y的形状为(3, 1, 4)。这样就在原来的Tensor中增加了一个新的维度。
需要注意的是,unsqueeze函数不会改变原始Tensor的形状,而是返回一个新的Tensor。
阅读全文