torch.broadcast_shapes

torch.broadcast_shapes(*shapes) → Size

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

這等同於 torch.broadcast_tensors(*map(torch.empty, shapes))[0].shape，但避免了創建中間張量的需求。這對於廣播具有常見批次形狀 (batch shape) 但最右側形狀不同的張量非常有用，例如，廣播具有共變異數矩陣的平均向量。

範例

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

參數

*shapes (torch.Size) – 張量的形狀。

回傳

與所有輸入形狀相容的形狀。

回傳類型

shape (torch.Size)

引發

RuntimeError – 如果形狀不相容。