Python PyTorch broadcast_shapes用法及代碼示例

本文簡要介紹python語言中 torch.broadcast_shapes 的用法。

用法:

torch.broadcast_shapes(*shapes) → Size

參數：

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

返回：

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

返回類型：

形狀(火炬.尺寸)

拋出：

RuntimeError - 如果形狀不兼容。

類似於 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])