torch.broadcast_tensors#
- torch.broadcast_tensors(*tensors) List of Tensors [source]#
将给定的张量根据 广播语义 进行广播。
- 参数
*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]])