Категориальная кроссентропия TFJS с использованием from_logits

Question

Я новичок в ML, и какое-то время боролся. Я конвертирую свой старый скрипт python в версию javascripttfjs. Однако я не могу найти одну строку в python, которая равна tfjs.

Я конвертирую скрипт Python:

model.compile(optimizer='adam',
  loss=tf.losses.CategoricalCrossentropy(from_logits=True),
  metrics=['accuracy'])

в скрипт javascript tfjs:

model.compile({
  optimizer: tf.train.adam(),
  loss: tf.losses.softmaxCrossEntropy,
  metrics: ['accuracy'],
});

Я не смог найти tfjs-версию tf.losses.CategoricalCrossentropy(from_logits=True) особенно from_logits переменная.

Любая помощь будет оценена по достоинству. Заранее спасибо.