Использование TF2 и преобразование тензора в массив путем вызова .numpy()
для него хорошо работает, но возвращает float32
, мне нужен массив numpy с более высокой точностью.
Прочитав даже источник Я не могу найти способ сделать это.
РЕДАКТИРОВАТЬ: Добавление некоторого кода, чтобы показать, как я собираю векторы.
model = tf.saved_model.load('./use_lite_saved_model')
sp = spm.SentencePieceProcessor()
sp.Load('./use_lite_saved_model/assets/universal_encoder_8k_spm.model')
def create_vector(sentence):
global sp
global model
values, indices, dense_shape = process_to_IDs_in_sparse_format(sp, [sentence])
tensors_out_dict = model.signatures['default'](
values=values,
indices=indices,
dense_shape=dense_shape
)
tensor_out = tensors_out_dict['default']
return tensor_out.numpy()[0]