Python PyTorch broadcast_to用法及代碼示例

本文簡要介紹python語言中 torch.broadcast_to 的用法。

用法:

torch.broadcast_to(input, shape) → Tensor

參數：

• input(Tensor) -輸入張量。

• shape(列表、元組或torch.Size) -新的形狀。

將 input 廣播到形狀 shape 。相當於調用 input.expand(shape) 。有關詳細信息，請參閱 expand() 。

例子：

>>> x = torch.tensor([1, 2, 3])
>>> torch.broadcast_to(x, (3, 3))
tensor([[1, 2, 3],
        [1, 2, 3],
        [1, 2, 3]])