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

Parameters

*形状 (torch.Size) – 张量的形状。

Returns

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

Return type

形状 (torch.Size)

Raises

RuntimeError – 如果形状不兼容。