当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch vander用法及代码示例


本文简要介绍python语言中 torch.vander 的用法。

用法:

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

参数

  • x(Tensor) -一维输入张量。

  • N(int,可选的) -输出中的列数。如果未指定 N,则返回一个方形数组

  • increasing(bool,可选的) -列的权力顺序。如果为真,则权力从左到右增加,如果为假(默认),则相反。

返回

范德蒙德矩阵。如果增加为 False,则第一列为 ,第二列为 ,依此类推。如果增加为 True,则列为

返回类型

Tensor

生成范德蒙德矩阵。

输出矩阵的列是输入向量 的元素幂。如果增加为 True,则列的顺序颠倒 。这种每行都有几何级数的矩阵被命名为Alexandre-Theophile Vandermonde。

例子:

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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.vander。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。