tfd_mse function

For using mean squared error via TFP