PyTorch Forums
trick