You could try doing the following:

Include the steps of forward_convolution_1 in the forward function of net, so that net(net_input) does not require any further processing before you pass it to mse.
Use torch.sqrt (or some combination of PyTorch functions) instead of np.sqrt.