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

Question

Коротко:

В Python версии Tensorflow существует метод tf.random.categorical () , который дает aws выборки из Категориальное распределение. Но я не могу найти подобный метод в 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 со случайным посевом, я думаю.