torch.broadcast_tensors

torch.broadcast_tensors(*tensors) → Tensor 列表[源代码]

按照广播语义广播给定的张量。

参数

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