如何在没有py_func的Tensorflow中使用python的“ord”函数

Question

在我的程序中，我需要根据ASCII表获取char的基数10值，例如，我通常会在python中获取x = ord('a')。

我有一个字符串占位符，我正在映射以获取每个字符，然后我需要返回等效的基数为10的数字。

鉴于我需要保存和恢复我的模型，我无法使用py_func来实现这一点，我找不到替代解决方案.r

Answer 1

使用tf.decode_raw：

import tensorflow as tf

a = tf.placeholder(dtype=tf.string, shape=())
number = tf.decode_raw(a, out_type=tf.uint8)

with tf.Session() as sess:
    print(sess.run(number, feed_dict={a: 'AB'}))  # prints [65 66]