U &ºcã@sŒddlZddlZddlZddlmmZe e¡Z dd„Z dd„Z dd„Z ej dkr\e ZnejZd d „Zeje eeje ed œZd d „ZdS)éNcCs|t |¡S)N)ÚtorchZsigmoid©Úx©rús&   ú