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

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

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

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

torch.squeeze(input, dim)

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

Пример

import torch

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

# Удалить все измерения размера 1
y = torch.squeeze(x)
print("Форма после squeeze:", y.shape)

# Указать размерность
z = torch.squeeze(x, dim=0)
print("После squeeze по dim=0:", z.shape)

Вывод:

原始形状: torch.Size([1, 2, 1, 3, 1])
挤压后形状: torch.Size([2, 3])
dim=0 挤压后: torch.Size([2, 1, 3, 1])

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

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