torch.broadcast_tensors

torch.broadcast_tensors(*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]])