torch.distributions.utils.broadcast_all

torch.distributions.utils.broadcast_all(*values) [исходный код]

Для заданного списка значений (возможно, содержащего числа) возвращает список, в котором каждое значение расширяется согласно следующим правилам:

• torch.*Tensor экземпляры расширяются согласно семантике расширения.
• Числовые экземпляры (скаляры) преобразуются в тензоры того же размера и типа, что и первый тензор, переданный в values. Если все значения являются скалярами, они преобразуются в скалярные тензоры.

Параметры:

values (список объектов Number, torch.*Tensor или объектов, реализующих __torch_function__) –

Исключения:

ValueError – если какое-либо из значений не является экземпляром Number, экземпляром torch.*Tensor или экземпляром, реализующим __torch_function__

Тип возвращаемого значения:

tuple[Tensor, …]