尝试将一些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()
非常感谢