torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size[源代码]

类似于broadcast_tensors()，但是针对形状。

这等同于torch.broadcast_tensors(*map(torch.empty, shapes))[0].shape，但避免了创建中间张量。这对于具有相同批次形状但不同最右形状的张量很有用，例如广播均值向量和协方差矩阵。

示例：

>>> torch.broadcast_shapes((2,), (3, 1), (1, 1, 1))
torch.Size([1, 3, 2])

参数

*shapes (torch.Size) — 表示张量的形状。

返回值

一个与所有输入形状都兼容的形状。

返回类型

形状（torch.Size）

异常

RuntimeError – 当形状不兼容时。