У меня есть обученная сеть CNN для набора данных MNIST в тензорном потоке JS. Я пытаюсь настроить вход так, чтобы один из выходных баллов был максимальным. Я знаю, как это сделать в python, используя такие keras:
from keras import backend as K
import numpy as np
class_num = 0
output = model.layers[-1].output
loss = K.mean(output[:,class_num])
grads = K.gradients(loss, model.input)[0]
grads = K.l2_normalize(grads)
func = K.function([model.input], [loss, grads])
input = np.random.random((1, 28, 28))
for i in range(10):
loss_val, grads_val = func([input])
input_img += grads_val
Проблема в том, что я не знаю, как сделать то же самое в тензорном потоке JS. Может ли кто-нибудь помочь в подобной реализации для tenorflow JS? Я не могу получить работающие функции потерь и градиента. Большое спасибо!