PyTorch Forums
xsquared