ttnn.transpose
- ttnn.transpose() None
-
Returns a tensor that is transposed along dims dim1 and dim2
Equivalent pytorch code:
output_tensor = torch.transpose(input_tensor, 0, 1)
:param *
input_tensor: Input Tensor. :param *dim1: First dim of transpose. :param *dim2: Second dim of transpose. :param *pad_value: padding value for when tiles are broken in a transpose. Defaults to 0.0. :type *pad_value: float, optional:keyword *
memory_config: Memory Config of the output tensor