Функция PyTorch torch.unsqueeze

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

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

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

torch.unsqueeze(input, dim)

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

Пример

import torch

x = torch.randn(2, 3)
print("Исходная форма:", x.shape)

Вставить измерение при dim=0
y = torch.unsqueeze(x, dim=0)
print("После расширения dim=0:", y.shape)

# Вставить измерение при dim=1
z = torch.unsqueeze(x, dim=1)
print("После расширения dim=1:", z.shape)

Вывод:

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

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

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