PyTorch torch.vander 函数

Pytorch torch 参考手册

torch.vander 是 PyTorch 中用于生成 Vandermonde 矩阵的函数。Vandermonde 矩阵是一种特殊矩阵，其每一行是输入向量的幂次。

函数定义

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

参数:

• x (Tensor): 输入一维张量。
• N (int, 可选): 输出矩阵的列数。默认为 len(x)。
• increasing (bool, 可选): 如果为 True，列的幂次递增；否则递减。默认为 False。

返回值:

• torch.Tensor: 返回 Vandermonde 矩阵。

使用示例

输出结果为：

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



实例 - 递增幂次

import torch



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



# 生成递增幂次的 Vandermonde 矩阵

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



print("递增 Vandermonde 矩阵:")

print(V_inc)





实例 - 指定列数

import torch



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



# 生成 5 列的 Vandermonde 矩阵

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



print("输入向量 x:", x)

print("nVandermonde 矩阵 (5列):")

print(V)





 Pytorch torch 参考手册
												
返回顶部