Mojo module
index_tensor
Functions
-
advanced_indexing_getitem
: Implement basic numpy-style advanced indexing. -
advanced_indexing_getitem_shape
: Calculate the output shape from advanced indexing. -
advanced_indexing_setitem_inplace
: Implement basic numpy-style advanced indexing with assignment. -
index_tensor
: Index_tensor operation; based on modified implementation of gather_nd. -
index_tensor_shape
: Compute the output shape of aindex_tensor
operation, and assert the inputs are compatible.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!