這篇主要就是用來建制之後android會用到的模組,首先我們要建立的是下圖的這個感知機
下方的code就是我們用來創建模組以及把他的圖像和變數儲存起來
import tensorflow as tf
I = tf.placeholder(tf.float32,shape=[None,3],name='I')
W = tf.Variable(tf.zeros(shape=[3,2]),dtype=tf.float32,name='W')
b = tf.Variable(tf.zeros(shape=[2]),dtype=tf.float32,name='b')
O = tf.nn.relu(tf.matmul(I,W)+b,name='O')
saver = tf.train.Saver()
init_op = tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init_op)
tf.train.write_graph(sess.graph_def, '.', 'android_tf.pbtxt')
sess.run(tf.assign(W,[[1,2],[3,4],[5,6]]))
sess.run(tf.assign(b,[1,2]))
saver.save(sess, 'android_tf.ckpt')