def mse(model_r, model_i, data_r, data_i, wgts):
    return tf.reduce_sum(((tf.square((data_r - model_r)) + tf.square((data_i - model_i))) * wgts))
def mse(model_r, model_i, data_r, data_i, wgts):
    return tf.reduce_sum(((tf.square((data_r - model_r)) + tf.square((data_i - model_i))) * wgts))