PyTorch torch.broadcast_to функция
Справочное руководство Pytorch torch
torch.broadcast_toЭто функция в PyTorch, используемая для трансляции тензора в заданную форму. Она возвращает представление входного тензора, транслированное в целевую форму. Правила трансляции следуют механизму трансляции NumPy.
Определение функции
torch.broadcast_to(input, shape)
Пример использования
Пример
import torch
# Базовое использование: трансляция одномерного тензора в двумерный
x = torch.tensor([1, 2, 3])
y = torch.broadcast_to(x, (3, 3))
print("Оригинал:", x)
print("После трансляции:")
print(y)
# tensor([[1, 2, 3],
# [1, 2, 3],
# [1, 2, 3]])
# Трансляция скаляра в форму большего размера
x = torch.tensor(5)
y = torch.broadcast_to(x, (2, 3, 4))
print("Скаляр транслирован в (2,3,4):", y.shape)
# Трансляция двумерного тензора в трёхмерный
x = torch.tensor([[1, 2], [3, 4]]) # (2, 2)
y = torch.broadcast_to(x, (3, 2, 2))
print("Трансляция в (3,2,2):", y.shape)
print(y)
# Базовое использование: трансляция одномерного тензора в двумерный
x = torch.tensor([1, 2, 3])
y = torch.broadcast_to(x, (3, 3))
print("Оригинал:", x)
print("После трансляции:")
print(y)
# tensor([[1, 2, 3],
# [1, 2, 3],
# [1, 2, 3]])
# Трансляция скаляра в форму большего размера
x = torch.tensor(5)
y = torch.broadcast_to(x, (2, 3, 4))
print("Скаляр транслирован в (2,3,4):", y.shape)
# Трансляция двумерного тензора в трёхмерный
x = torch.tensor([[1, 2], [3, 4]]) # (2, 2)
y = torch.broadcast_to(x, (3, 2, 2))
print("Трансляция в (3,2,2):", y.shape)
print(y)
Другие расширения