Mojo function
slice_shape
slice_shape[input_type: DType, start_type: DType, stop_type: DType, step_type: DType](input_buf: TileTensor[input_type, input_buf.LayoutType, input_buf.origin, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_size=input_buf.element_size], start_buf: TileTensor[start_type, start_buf.LayoutType, start_buf.origin, address_space=start_buf.address_space, linear_idx_type=start_buf.linear_idx_type, element_size=start_buf.element_size], stop_buf: TileTensor[stop_type, stop_buf.LayoutType, stop_buf.origin, address_space=stop_buf.address_space, linear_idx_type=stop_buf.linear_idx_type, element_size=stop_buf.element_size], step_buf: TileTensor[step_type, step_buf.LayoutType, step_buf.origin, address_space=step_buf.address_space, linear_idx_type=step_buf.linear_idx_type, element_size=step_buf.element_size]) -> IndexList[TileTensor[input_type, input_buf.LayoutType, input_buf.origin, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_size=input_buf.element_size].rank]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!