torch.linalg.vander¶
- torch.linalg.vander(x, N=None) Tensor ¶
生成范德蒙德矩阵。
返回范德蒙德矩阵
对于 N > 1。如果
N
= None,则 N = x.size(-1),使得输出为一个方阵。支持 float, double, cfloat, cdouble 和整数 dtype 的输入。也支持向量的批次,如果
x
是一批向量,则输出具有相同的批次维度。与 numpy.vander 的区别
与 numpy.vander 不同,此函数按升序返回
x
的幂。要按逆序获取它们,请调用linalg.vander(x, N).flip(-1)
。
- 参数
x (Tensor) – 形状为 (*, n) 的张量,其中 * 为零或多个由向量组成的批次维度。
- 关键字参数
N (int, optional) – 输出中的列数。默认值: 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]])