def test_lexsort_two_dim(self, xp):
a = xp.array([[9, 4, 0, 4, 0, 2, 1],
[1, 5, 1, 4, 3, 4, 4]]) # from numpy.lexsort example
return xp.lexsort(a)
微信
公众号
扫码关注公众号