Регуляризация функции стоимости в TensorFlow.js

Мне интересно, может ли кто-нибудь помочь мне. Я новичок в TensorFlow.js (версия JavaScript). Я построил нейронную сеть и хочу добавить член регуляризации в функцию стоимости (функцию потерь).

Я вижу регуляризаторы в документации JavaScript API, но не могу понять, как их использовать. Со слоями может быть связан какой-то регуляризатор, но функция стоимости не определена в слоях, поэтому я не думаю, что это то, что я ищу.

Я просмотрел исходный код на GitHub. Я нашел несколько открытых тикетов, в которых упоминалось о регуляризации. Я также нашел функцию регуляризации, которая применяла норму L2 или L1 к вектору. Я могу попытаться написать функцию, которая увеличивает функцию стоимости, используя функцию регуляризации, но я не хочу прилагать столько усилий, когда функция уже существует. Версия TensorFlow для Python содержит то, что я ищу. Кто-нибудь знает, существует ли то, что я ищу, в версии javascript, и если да, то как мне это реализовать? Спасибо.