tensorflow 0.21.0

Rust language bindings for TensorFlow.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

import tensorflow as tf

# check tensorflow version is 2.x 
tf_major_version = tf.__version__.split('.')[0]
assert tf_major_version == '2'

@tf.function
def add(x, y):
    tf.add(x, y, name='z')

x = tf.TensorSpec((), dtype=tf.dtypes.int32, name='x')
y = tf.TensorSpec((), dtype=tf.dtypes.int32, name='y')

concrete_function = add.get_concrete_function(x, y)
directory = 'examples/addition'
tf.io.write_graph(concrete_function.graph, directory, 'model.pb', as_text=False)