PyTorch Forums
a_bear_paw