def my_rmse(y_true, y_pred):
    """RMSE calculation using keras.backend"""
    from keras import backend as kb
    sq_err = kb.square(y_pred - y_true)
    mse = kb.mean(sq_err, axis=-1)
    rmse = kb.sqrt(mse)
    return rmse
