torch.broadcast_tensors

torch.broadcast_tensors(*tensors) → 列表中的张量[source][source] ¶

根据广播语义广播给定的张量。

参数:

*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]])