torch 中的别名

创建于：2025 年 7 月 22 日 | 最后更新于：2025 年 7 月 22 日

以下是 torch 中与其定义所在嵌套命名空间中的对应项的别名。您可以随意使用 torch 中的顶层版本（例如 torch.broadcast_tensors()）或嵌套版本 torch.functional.broadcast_tensors()。

align_tensors
atleast_1d	返回每个输入张量的一维视图，其中零维度。
atleast_2d	为每个输入张量返回一个具有零维度的二维视图。
atleast_3d	为每个输入张量返回一个具有零维度的三维视图。
block_diag	从提供的张量创建块对角矩阵。
broadcast_shapes	类似于 broadcast_tensors()，但用于形状。
broadcast_tensors	根据 广播语义 广播给定的张量。
cartesian_prod	对给定的张量序列进行笛卡尔积。
cdist	计算两个行向量集合之间每对向量的 p-范数距离（批处理）。
chain_matmul	返回 $N$ 个二维张量的矩阵乘积。
einsum	使用基于爱因斯坦求和约定的符号，对输入 operands 的元素在指定维度上的乘积进行求和。
lu	计算矩阵或矩阵批次 A 的 LU 分解。
meshgrid	创建由 attr:tensors 中一维输入指定的坐标网格。
norm	返回给定张量的矩阵范数或向量范数。
split	将张量分割成块。
stft	短时傅里叶变换 (STFT)。
tensordot	返回 a 和 b 在多个维度上的收缩。
unique	返回输入张量的唯一元素。
unique_consecutive	从每个连续的等效元素组中删除除第一个元素之外的所有元素。
unravel_index	将平坦索引张量转换为坐标张量元组，这些元组可以索引到任意指定形状的张量。