PyTorch torch.broadcast_tensors функция
Справочное руководство Pytorch torch
torch.broadcast_tensorsЭто функция в PyTorch, используемая для широковещания нескольких тензоров. Она выполняет широковещание входных тензоров до одной формы и возвращает набор тензоров, с которыми можно выполнять поэлементные операции. Правила широковещания следуют механизму широковещания NumPy.
Определение функции
torch.broadcast_tensors(*tensors)
Примеры использования
Пример
import torch
# Трансляция тензоров разных форм
x = torch.tensor([[1, 2, 3]]) # Форма (1, 3)
y = torch.tensor([[1], [2], [3]]) # Форма (3, 1)
a, b = torch.broadcast_tensors(x, y)
print("форма x после трансляции:", a.shape)
print("форма y после трансляции:", b.shape)
print("x после трансляции:")
print(a)
print("y после трансляции:")
print(b)
print("Поэлементная сумма:")
print(a + b)
# Несколько тензоров
x1 = torch.randn(3, 1, 5)
x2 = torch.randn(1, 4, 5)
x3 = torch.randn(3, 4, 1)
r1, r2, r3 = torch.broadcast_tensors(x1, x2, x3)
print("Форма после трансляции:", r1.shape)
# Трансляция тензоров разных форм
x = torch.tensor([[1, 2, 3]]) # Форма (1, 3)
y = torch.tensor([[1], [2], [3]]) # Форма (3, 1)
a, b = torch.broadcast_tensors(x, y)
print("форма x после трансляции:", a.shape)
print("форма y после трансляции:", b.shape)
print("x после трансляции:")
print(a)
print("y после трансляции:")
print(b)
print("Поэлементная сумма:")
print(a + b)
# Несколько тензоров
x1 = torch.randn(3, 1, 5)
x2 = torch.randn(1, 4, 5)
x3 = torch.randn(3, 4, 1)
r1, r2, r3 = torch.broadcast_tensors(x1, x2, x3)
print("Форма после трансляции:", r1.shape)
Другие расширения