Función torch.permute de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.permuteEs una función de PyTorch utilizada para reordenar las dimensiones de un tensor.

Definición de la función

torch.permute(input, dims)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor 3D
x = torch.randn(2, 3, 4)
print(Forma original:, x.shape)

# Reordenar dimensiones (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("después de permute:", y.shape)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones