Преобразовать тензор в массив numpy, но с большей точностью - PullRequest
1 голос
/ 03 февраля 2020

Использование 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]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...