chectus/chectus_net/cpe2wdl.py
2024-07-28 14:48:02 +08:00

7 lines
80 B
Python

import torch
scale = 410
def cpe2wdl(cpe):
return torch.sigmoid(cpe / scale)