torch.linalg.vander

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

Генерирует матрицу Вандермонда.

Возвращает матрицу Вандермонда $V$

$$V = \begin{pmatrix} 1 & x_1 & x_1^2 & \dots & x_1^{N-1}\\ 1 & x_2 & x_2^2 & \dots & x_2^{N-1}\\ 1 & x_3 & x_3^2 & \dots & x_3^{N-1}\\ \vdots & \vdots & \vdots & \ddots &\vdots \\ 1 & x_n & x_n^2 & \dots & x_n^{N-1} \end{pmatrix}.$$

для N > 1. Если N= None, то N = x.size(-1) для того, чтобы выход был квадратной матрицей.

Поддерживает входные данные типов float, double, cfloat, cdouble и целочисленных dtypes. Также поддерживает пакеты векторов, и если x является пакетом векторов, то выход имеет те же размерности пакета.

Отличия от numpy.vander:

• В отличие от numpy.vander, эта функция возвращает степени x в возрастающем порядке. Чтобы получить их в обратном порядке, вызовите linalg.vander(x, N).flip(-1).

Параметры:

x (Тензор) – тензор формы (*, n) где * — это ноль или более размерностей пакета, состоящих из векторов.

Аргументы по ключевым словам:

N (int, необязательно) – Количество столбцов в выходе. Значение по умолчанию: x.size(-1)

Пример:

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