It accepts an ndarray with dimensions (..., m, m). Not sure what .tolist() does, but if you wrap it in np.asarray and make sure the core dims are the rightmost ones, it will work.
1 Like
It accepts an ndarray with dimensions (..., m, m). Not sure what .tolist() does, but if you wrap it in np.asarray and make sure the core dims are the rightmost ones, it will work.