We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
axis
I suspect take's axis argument will be needed at some point. Can we add a simple implementation for PyTorch?
def take(array, indices, *, axis): key = [slice(None)] * array.ndim key[axis] = indices return array[key]