Функция PyTorch torch.unbind
Справочное руководство Pytorch torch
torch.unbindЭто функция PyTorch, используемая для разделения тензора на кортеж вдоль указанного измерения.
Определение функции
torch.unbind(input, dim=0)
Пример использования
Пример
import torch
x = torch.arange(12).reshape(3, 4)
print(Исходный тензор:)
print(x)
result = torch.unbind(x, dim=0)
print(Разделение по dim=0:)
for i, t in enumerate(result):
print(fБлок {i}: {t})
result = torch.unbind(x, dim=1)
print(Разделение по dim=1:)
for i, t in enumerate(result):
print(fБлок {i}: {t})
x = torch.arange(12).reshape(3, 4)
print(Исходный тензор:)
print(x)
result = torch.unbind(x, dim=0)
print(Разделение по dim=0:)
for i, t in enumerate(result):
print(fБлок {i}: {t})
result = torch.unbind(x, dim=1)
print(Разделение по dim=1:)
for i, t in enumerate(result):
print(fБлок {i}: {t})
Результат вывода:
原始张量:
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
沿 dim=0 分割:
块 0: tensor([0, 1, 2, 3])
Chunk 1: tensor([4, 5, 6, 7])
Chunk 2: tensor([ 8, 9, 10, 11])
沿 dim=1 分割:
块 0: tensor([0, 4, 8])
Chunk 1: tensor([1, 5, 9])
Chunk 2: tensor([ 2, 6, 10])
Chunk 3: tensor([ 3, 7, 11])
Другие расширения