numpy的类似unsqueeze
时间: 2023-10-16 12:00:29 浏览: 119
在NumPy中,可以使用`np.newaxis`来增加数组的维度,类似于PyTorch中的`unsqueeze`操作。通过将`np.newaxis`放在索引位置上,可以在该位置上增加一个新的维度。例如,假设有一个数组`a`,可以使用`a\[np.newaxis, :\]`来在第一个维度上增加一个新的维度。这将使得原来的一维数组变成一个二维数组,形状为`(1, 3)`。\[1\]
#### 引用[.reference_title]
- *1* *2* [numpy数组操作](https://blog.csdn.net/weixin_41978699/article/details/122441895)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [torch和numpy的各种常用的矩阵matrix操作](https://blog.csdn.net/XC___XC/article/details/121442494)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文