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])
- 参数
*shapes (torch.Size) – 张量的形状。
- 返回值
与所有输入形状兼容的形状。
- 返回类型
shape (torch.Size)
- 引发异常
RuntimeError – 如果形状不兼容。