Я хочу сохранить некоторые Bert Ins для быстрого поиска в обучении тензорного потока, которые будут использоваться для входных данных в обучении Bert. Однако кажется, что слои встраивания keras не определяют тип данных, поэтому, вероятно, это только float32.
https://www.tensorflow.org/api_docs/python/tf/keras/layers/Embedding
Есть ли способ получить int16 или вложения int32?