VarList, переданный в variableGrads(f, varList), должен быть массивом переменных Tensorflow.js

Question

У меня есть этот код, который предназначен для оптимизации весов моих нейронных сетей, и я продолжаю получать эту ошибку:

VarList, переданный в variableGrads(f, varList), должен быть массивом переменных.

Код:

var Wax = parameters["Wax"]
var Waa = parameters["Waa"]
var ba = parameters["ba"]
var Wya = parameters["Wya"]
var by = parameters["by"]


var model = RNN_FowardProp(X, a0, parameters)
let loss = tf.losses.meanSquaredError (Y, model)
const f = () => loss

tf.train.sgd(0.01).minimize(f, false, [Wax, Waa, ba, Wya, by])