Skip to main content

Mojo function

layout_tensor_reshape

layout_tensor_reshape[output_rank: Int, dtype: DType, single_thread_blocking_override: Bool](input: LayoutTensor[dtype, layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], new_shape: IndexList[output_rank]) -> LayoutTensor[dtype, Layout.row_major[output_rank](), origin, address_space=address_space]

Returns:

LayoutTensor

Was this page helpful?