Python PyTorch frac用法及代碼示例

本文簡要介紹python語言中 torch.frac 的用法。

用法:

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

計算 input 中每個元素的小數部分。

\text{out}_{i} = \text{input}_{i} - \left\lfloor |\text{input}_{i}| \right\rfloor * \operatorname{sgn}(\text{input}_{i})

例子：

>>> torch.frac(torch.tensor([1, 2.5, -3.2]))
tensor([ 0.0000,  0.5000, -0.2000])