Функция torch.permute в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.permuteявляется функцией в PyTorch, используемой для перестановки размерностей тензора.

Определение функции

torch.permute(input, dims)

Примеры использования

Пример

import torch

# Создать 3D-тензор
x = torch.randn(2, 3, 4)
print("Исходная форма:", x.shape)

# Переставить размерности (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("После permute:", y.shape)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения