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