can you provide this code in pytorch .thanks
can you provide this code in pytorch .thanks