torch.vander

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

產生 Vandermonde 矩陣。

輸出矩陣的各列是輸入向量的逐元素冪 $x^{(N-1)}, x^{(N-2)}, ..., x^0$。 如果 increasing 為 True，則列的順序會反轉 $x^0, x^1, ..., x^{(N-1)}$。 這種每行都是等比數列的矩陣以 Alexandre-Theophile Vandermonde 命名。

參數

• x (Tensor) – 1 維輸入張量。

• N (int, optional) – 輸出中的列數。 如果未指定 N，則會傳回一個方陣 $(N = len(x))$。

• increasing (bool, optional) – 列的冪的順序。 如果為 True，則冪從左到右遞增；如果為 False（預設），則會反轉。

返回值

Vandermonde 矩陣。 如果 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]])