Как рисовать образцы из категориального распределения в TensorFlow.js

Question

Вкратце о проблеме

В версии Tensorflow для Python есть метод tf.random.categorical(), который извлекает образцы из категориального распределения. Но я не могу найти аналогичный метод в TensorFlow.js API. Итак, как правильно рисовать образцы из категориального распределения в TensorFlow.js?

Подробнее о проблеме

В генерации текста с помощью учебника RNN метод tf.random.categorical() используется вgenerate_text() функция, чтобы решить, какой символ следует передать рядом с входом RNN для создания последовательности.

predicted_id = tf.random.categorical(predictions, num_samples=1)[-1,0].numpy()

Я экспериментирую с TensorFlow.js и пытаюсь создать "случайное" письмо, подобное Шекспиру, но в браузере. Кажется, что все части учебника хорошо работают вместе, кроме шага с использованиемtf.random.categorical() метод.

Я думаю, что пишу альтернативу tf.random.categorical() функция вручную не должна быть такой сложной, а также есть несколько сторонних библиотек JavaScript, которые уже реализуют эту функцию, но выглядит довольно логичным, чтобы иметь ее как часть TensorFlow.js API.

Answer 1

Я думаю ты можешь использовать tf.multinomial вместо.

Я заглянул в исходный код и сname а также seed для параметров установлено значение None, это практически то же самое, что tf.multinomial я полагаю, с некоторым случайным посевом.