类型错误：numpy 中无法理解数据类型

Question

我正在使用 numpy 进行神经网络工作，但遇到了这个错误

类型错误：数据类型无法理解

我的代码是

a = np.array([7, 7, 7], [7, 7, 7])

print(np.argmax(np.array([[5, 6, 7], [1, 2, 3]]), axis = 0, a))

Answer 1

 np.array 接收两个输入，第一个是数据本身，第二个是数据类型。当您想要两行时，您应该为 np.array 提供一个列表列表。np.array 尝试创建一个包含元素 [7, 7, 7] 的一维数组，而另一个 [7, 7, 7] 被认为是 dtype，这导致了错误。