AttributeError: 'numpy.ndarray' object has no attribute 'unsqueeze'怎么解决

这个错误是因为 NumPy 中的 ndarray 对象没有 unsqueeze() 方法。

如果您想将 ndarray 转换为 PyTorch 张量并使用 unsqueeze() 方法，请使用以下代码：

import torch
import numpy as np

arr = np.array([[1,2,3], [4,5,6]])
tensor = torch.from_numpy(arr).unsqueeze(0) # 在第 0 维插入一个新的维度

如果您只是想在 NumPy 中添加一个新维度，请使用以下代码：

import numpy as np

arr = np.array([[1,2,3], [4,5,6]])
new_arr = np.expand_dims(arr, axis=0) # 在第 0 维插入一个新的维度