search

data-apis / array-api-tests

Test suite for the PyData Array APIs standard
https://data-apis.org/array-api-tests/
MIT License
64 stars 41 forks source link

Avoid passing a list of axes to tensordot #237

Closed asmeurer closed 8 months ago

asmeurer commented 8 months ago

cupy has logic that fails if axis is not a tuple. The standard type annotations only require axes to be a tuple.