快捷方式

torch.vander

torch.vander(x, N=None, increasing=False) Tensor

生成范德蒙矩阵。

输出矩阵的列是输入向量 x(N1),x(N2),...,x0x^{(N-1)}, x^{(N-2)}, ..., x^0的逐元素幂。如果 increasing 为 True,则列的顺序将反转 x0,x1,...,x(N1)x^0, x^1, ..., x^{(N-1)}。这种在每一行都有几何级数的矩阵以亚历山大-特奥菲尔·范德蒙命名。

参数
  • x (张量) – 一维输入张量。

  • N (int, 可选) – 输出中的列数。如果未指定 N,则返回一个方阵 (N=len(x))(N = len(x)).

  • increasing (bool, 可选) – 列的幂的顺序。如果为 True,则幂从左到右递增,如果为 False(默认值),则反转。

返回

范德蒙德矩阵。如果 increasing 为 False,则第一列为 x(N1)x^{(N-1)}, 第二列为 x(N2)x^{(N-2)},以此类推。如果 increasing 为 True,则各列为 x0,x1,...,x(N1)x^0, x^1, ..., x^{(N-1)}.

返回类型

张量

示例

>>> x = torch.tensor([1, 2, 3, 5])
>>> torch.vander(x)
tensor([[  1,   1,   1,   1],
        [  8,   4,   2,   1],
        [ 27,   9,   3,   1],
        [125,  25,   5,   1]])
>>> torch.vander(x, N=3)
tensor([[ 1,  1,  1],
        [ 4,  2,  1],
        [ 9,  3,  1],
        [25,  5,  1]])
>>> torch.vander(x, N=3, increasing=True)
tensor([[ 1,  1,  1],
        [ 1,  2,  4],
        [ 1,  3,  9],
        [ 1,  5, 25]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源