PyTorch Forums
function2