Python PyTorch vander用法及代碼示例

本文簡要介紹python語言中 torch.vander 的用法。

用法:

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

參數：

• x(Tensor) -一維輸入張量。

• N(int,可選的) -輸出中的列數。如果未指定 N，則返回一個方形數組 (N = len(x)) 。

• increasing(bool,可選的) -列的權力順序。如果為真，則權力從左到右增加，如果為假(默認)，則相反。

返回：

範德蒙德矩陣。如果增加為 False，則第一列為 x^{(N-1)} ，第二列為 x^{(N-2)} ，依此類推。如果增加為 True，則列為 x^0, x^1, ..., x^{(N-1)} 。

返回類型：

Tensor

生成範德蒙德矩陣。

輸出矩陣的列是輸入向量 x^{(N-1)}, x^{(N-2)}, ..., x^0 的元素冪。如果增加為 True，則列的順序顛倒 x^0, x^1, ..., x^{(N-1)} 。這種每行都有幾何級數的矩陣被命名為Alexandre-Theophile Vandermonde。

例子：

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