torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size

與 broadcast_tensors() 類似，但用於形狀（shapes）。

這等同於 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 – 如果形狀不相容。