快捷方式

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 – 如果形状不兼容。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源