numpy 中哪个函数类似于unsqueeze
时间: 2024-05-03 18:17:56 浏览: 129
详解pytorch中squeeze()和unsqueeze()函数介绍
在numpy中,可以使用`numpy.newaxis`来扩展数组的维度,类似于`unsqueeze`函数。例如,假设我们有一个形状为`(3, 4)`的数组`arr`,我们可以通过以下方式在第一维上添加一个新的维度:
```python
import numpy as np
arr = np.zeros((3, 4))
new_arr = arr[:, np.newaxis, :]
print(new_arr.shape) # 输出 (3, 1, 4)
```
这里我们使用了`newaxis`来在第二个维度上插入一个新的维度。同样的,我们也可以在其他维度上插入新的维度。
阅读全文