PyTorch Forums
TO_Kocsis