2017-02-05 3 views

答えて

0

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をご確認ください。

関連する問題