Función torch.vander de PyTorch

Pytorch torch 参考手册 Manual de referencia de PyTorch torch

torch.vanderEs una función en PyTorch utilizada para generar la matriz de Vandermonde. La matriz de Vandermonde es una matriz especial cuyas filas son potencias del vector de entrada.

Definición de la función

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

Parámetros:

x(Tensor): tensor unidimensional de entrada.
N(int, opcional): número de columnas de la matriz de salida. Por defecto es len(x).
increasing(bool, opcional): si es True, las potencias de las columnas aumentan; de lo contrario, disminuyen. Por defecto es False.

Valor de retorno:

torch.Tensor: devuelve la matriz de Vandermonde.

Ejemplo de uso

Ejemplo

import torch

# Crear el vector de entrada
x = torch.tensor([1, 2, 3])

# Generar la matriz de Vandermonde
V = torch.vander(x)

print("Vector de entrada x:", x)
print("nMatriz de Vandermonde:")
print(V)

El resultado de salida es:

输入向量 x: tensor([1, 2, 3])
Vandermonde 矩阵:
tensor([[1, 1, 1],
        [4, 2, 1],
        [9, 3, 1]])



Ejemplo - Potencias ascendentes

import torch



x = torch.tensor([1, 2, 3])



# Generar la matriz de Vandermonde con potencias ascendentes

V_inc = torch.vander(x, increasing=True)



print("Matriz de Vandermonde ascendente:")

print(V_inc)





Ejemplo - Especificar número de columnas

import torch



x = torch.tensor([1, 2, 3, 4])



# Generar una matriz de Vandermonde de 5 columnas

V = torch.vander(x, N=5)



print("Vector de entrada x:", x)

print("nMatriz de Vandermonde (5 columnas):")

print(V)





Manual de referencia de PyTorch torch