当前位置: 首页>>编程示例 >>用法及示例精选 >>正文


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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.broadcast_to。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。