python - 如何将 numpy.float64 数组转换为 float64 数组？

Question

我有一个包含其他几个 np.array 的 np.array，它们本身包含 numpy.float64 类型的对象，这导致我的一种方法出现以下问题：

TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.

错误就在我的代码的这一行上：

self.x = torch.from_numpy(X_train)[train]

有没有办法将 np.array 的内容转换为常规类型而不是 numpy 对象？

Answer 1

好的，我进行了调查，这是我的数组元素之一的维度问题。谢谢大家的回答。