Mojo function
transpose_helper
transpose_helper[dtype: DType](output_tensor: TileTensor[dtype, origin, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], input_tensor: TileTensor[dtype, origin, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], ctx: DeviceContext)
Helper function to transpose a tensor from [B, N, K] to [N, B, K] (or vice versa).
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!