https://bestthaifoodrecipeshopppp.blogspot.com/2022/08/custom-layer-from-keras-to-pytorch.html