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

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

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

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

torch.flatten(input, start_dim, end_dim, out)

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

Пример

import torch

x = torch.randn(2, 3, 4)

# Уплощение до одного измерения
y = torch.flatten(x)

print("Исходная форма:", x.shape)
print("После уплощения:", y.shape)

# Сглаживание из указанного измерения
z = torch.flatten(x, start_dim=1)
print("Сглаживание из dim=1:", z.shape)

Результат вывода:

原始形状: torch.Size([2, 3, 4])
展平后: torch.Size([24])
从 dim=1 展平: torch.Size([2, 12])

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

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