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