Function style_loss

cnn_class2/style_transfer2.py:42–43

(y, t)

def style_loss(y, t):
  return K.mean(K.square(gram_matrix(y) - gram_matrix(t)))


# let's generalize this and put it into a function