返回ndarray的Numpy拆分

有没有类似的方法 numpy.split 返回一个 numpy.ndarray 而不是 list ?假设数组平均分割是可以的(以防止出现锯齿状数组)。

例如:

x = np.arange(9.0)
print(np.split(x, 3))
# [array([0.,  1.,  2.]), array([3.,  4.,  5.]), array([6.,  7.,  8.])]

print(np.???(x, 3))
# array([[0.,  1.,  2.], [3.,  4.,  5.], [6.,  7.,  8.]])

我不想把给出的清单叠在一起 np.split 出于性能原因。