torch.broadcast_tensors

torch.broadcast_tensors(*tensors) → List of Tensors [исходный код]

Расширяет заданные тензоры согласно семантике расширения.

Параметры:

*tensors – любое количество тензоров одного типа

Предупреждение

Несколько элементов расширенного тензора могут ссылаться на одну и ту же ячейку памяти. В результате операции на месте (особенно векторизованные) могут привести к некорректному поведению. Если вам нужно записывать данные в тензоры, сначала клонируйте их.

Пример:

>>> x = torch.arange(3).view(1, 3)
>>> y = torch.arange(2).view(2, 1)
>>> a, b = torch.broadcast_tensors(x, y)
>>> a.size()
torch.Size([2, 3])
>>> a
tensor([[0, 1, 2],
        [0, 1, 2]])