torch.ldexp

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

将 input 乘以 2 ** other。

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

通常此函数用于通过将 input 中的尾数乘以由 other 中的指数创建的 2 的整数次幂来构建浮点数。

参数

• input (张量) – 输入张量。

• other (张量) – 指数张量，通常为整数。

关键字参数

out (张量, 可选) – 输出张量。

示例

>>> 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.])