torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size

类似于 broadcast_tensors()，但用于 shape。

这等效于 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) – 张量的形状。

返回

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

返回类型

shape (torch.Size)

引发

RuntimeError – 如果形状不兼容。