torch.vander

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

生成一个范德蒙德矩阵。

输出矩阵的列是输入向量 $x^{(N-1)}, x^{(N-2)}, ..., x^0$) 的逐元素幂。如果 increasing 为 True，则列的顺序反转为 $x^0, x^1, ..., x^{(N-1)}$。每行具有几何级数（等比数列）的此类矩阵以 Alexandre-Theophile Vandermonde 命名。

参数

• x (Tensor) – 1-D 输入张量。

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

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

返回值

范德蒙德矩阵。如果 increasing 为 False，第一列是 $x^{(N-1)}$，第二列是 $x^{(N-2)}$，依此类推。如果 increasing 为 True，列是 $x^0, x^1, ..., x^{(N-1)}$。

返回类型

Tensor

示例

>>> 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]])