快捷方式

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源