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

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

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

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

torch.split(tensor, split_size, dim)

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

Пример

import torch

x = torch.arange(12).reshape(3, 4)

# Разделить по dim=0
result = torch.split(x, 1, dim=0)

for i, t in enumerate(result):
    print(f"Часть {i}:")
    print(t)

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

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