Функция torch.chunk в PyTorch
Справочное руководство Pytorch torch
torch.chunkЭто функция в PyTorch, используемая для разделения тензора на части по указанному измерению.
Определение функции
torch.chunk(tensor, chunks, dim)
Пример использования
Пример
import torch
x = torch.arange(12).reshape(3, 4)
# Разделить на 3 части
result = torch.chunk(x, 3, dim=0)
print("Количество частей:", len(result))
for i, t in enumerate(result):
print(f"Часть {i}:", t.shape)
x = torch.arange(12).reshape(3, 4)
# Разделить на 3 части
result = torch.chunk(x, 3, dim=0)
print("Количество частей:", len(result))
for i, t in enumerate(result):
print(f"Часть {i}:", t.shape)
Другие расширения