微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python中tf.train.Example.SerializeToString的Tensorflow Java API是什么?

尝试将一些python tensorflow代码转换为Java,我在python中找不到相同的java API for train.example.SerializetoString。它肯定不是java toString。我可以在https://www.javadoc.io/doc/org.tensorflow/proto/latest/org/tensorflow/example/package-summary.html处找到java Example类,但是根本没有诸如SerializetoString这样的API。

在Python中

import numba as nb
from numba.typed import List

@nb.njit
def number_of_unique_lists(a):
    s = List()
    for x in a:
        if not x in s:
            s.append(x)
    number = len(s)
    return number

a = [[1,2,3],[1,2],[3,4],3]]
typed_a = List()
for x in a:
    s = List()
    for y in x:
        s.append(y)
    typed_a.append(s)
print(number_of_unique_lists(typed_a))
>>> 3

我可以转换成Java

features["a"] = tf.train.Feature(int64_list=tf.train.Int64List(value=list(values_a)))
features["b"] = tf.train.Feature(int64_list=tf.train.Int64List(value=list(values_b)))
tf_example = tf.train.Example(features=tf.train.Features(feature=features))
model_input = tf_example.SerializetoString()

非常感谢

原文地址:https://www.jb51.cc/python/3189340.html

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐