Skip to main content

Mojo function

slice_dim_as_view

slice_dim_as_view[dtype: DType, dim: Int](tensor: TileTensor[dtype, tensor.LayoutType, tensor.origin, address_space=tensor.address_space, linear_idx_type=tensor.linear_idx_type, element_size=tensor.element_size], start: Int, end: Int, step: Int) -> TileTensor[dtype, Layout[*?, *?], tensor.origin, address_space=tensor.address_space]

Returns:

TileTensor

Was this page helpful?