1
luaからtensorflowへのコードを実装しようとしています。 tensorflowにおけるnn.Linear()の同様の関数は何ですか?同じ結果を得るための置換や類似の関数。関数はテンソルフローのLua nn.Linear()と似ています
luaからtensorflowへのコードを実装しようとしています。 tensorflowにおけるnn.Linear()の同様の関数は何ですか?同じ結果を得るための置換や類似の関数。関数はテンソルフローのLua nn.Linear()と似ています
にslim.fully_connected
をちょうど使用してください。トーチで
module= nn.Linear(input_dims, output_dims)
は、データのバッチopertionについて考慮しないtensorflowで
input = tf.placeholder(tf.float32, [batch_size] + input_dims, name='real_images')
output = slim.fully_connected(input, output_dims, activation_fn=None)
と同等です。
tf.contrib.slim documentationをご確認ください。