Python PyTorch broadcast_tensors用法及代碼示例

本文簡要介紹python語言中 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]])