Fix the torch.take() wrapper to make axis optional for ndim = 1 - Githubissues

data-apis / array-api-compat

Compatibility layer for common array libraries to support the Array API

https://data-apis.org/array-api-compat/

MIT License

69 stars 22 forks source link

Fix the torch.take() wrapper to make axis optional for ndim = 1 #47

Closed asmeurer closed 1 year ago

asmeurer commented 1 year ago

Closes #34