Python PyTorch ldexp用法及代码示例

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

用法:

torch.ldexp(input, other, *, out=None) → Tensor

参数：

• input(Tensor) -输入张量。

• other(Tensor) - index 的张量，通常是整数。

关键字参数：

out(Tensor,可选的) -输出张量。

将 input 乘以 2**:attr: other 。

\text{{out}}_i = \text{{input}}_i * 2^\text{{other}}_i

通常，此函数用于通过将 input 中的尾数乘以从 other 中的 index 创建的 2 的整数幂来构造浮点数。

例子：

>>> torch.ldexp(torch.tensor([1.]), torch.tensor([1]))
tensor([2.])
>>> torch.ldexp(torch.tensor([1.0]), torch.tensor([1, 2, 3, 4]))
tensor([ 2.,  4.,  8., 16.])