torch.linalg.vander¶
- torch.linalg.vander(x, N=None) Tensor ¶
生成范德蒙矩阵。
返回范德蒙矩阵
对于 N > 1。如果
N
= None,则 N = x.size(-1),以便输出为方阵。支持浮点型、双精度浮点型、复数浮点型、复数双精度浮点型和整型数据类型。也支持向量批处理,如果
x
是向量批处理,则输出具有相同的批处理维度。与 numpy.vander 的区别
与 numpy.vander 不同,此函数按升序返回
x
的幂。要按降序获取,请调用linalg.vander(x, N).flip(-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]])