def style_loss(Y_hat, gram_Y):
    return np.square(gram(Y_hat) - gram_Y).mean()