Mojo function
gather_guards
gather_guards(axis: Axis, input_shape: IndexList[input_shape.size, element_type=input_shape.element_type], indices_shape: IndexList[indices_shape.size, element_type=indices_shape.element_type], output_shape: IndexList[output_shape.size, element_type=output_shape.element_type])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!