torch_transpose: Transpose

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Transpose

Description

Transpose

Usage

torch_transpose(self, dim0, dim1)

Arguments

self	(Tensor) the input tensor.
dim0	(int) the first dimension to be transposed
dim1	(int) the second dimension to be transposed

transpose(input, dim0, dim1) -> Tensor

Returns a tensor that is a transposed version of input. The given dimensions dim0 and dim1 are swapped.

The resulting out tensor shares it's underlying storage with the input tensor, so changing the content of one would change the content of the other.

Examples

if (torch_is_installed()) {

x = torch_randn(c(2, 3))
x
torch_transpose(x, 1, 2)
}

torch documentation built on May 29, 2024, 9:54 a.m.