преобразующий выходные данные слоя Softmax в категориальный массив?

#tensorflow #keras Вопрос: Мне нужен слой, который преобразует softmax в категориальный вектор (а также выбирает категорию "случайным образом", используя вероятность softmax), вот мой код на данный момент: from tensorflow import…