函数类似于张量流中的Lua nn.Linear（）

Question

我正在尝试从lua到tensorflow实现代码。张量流中nn.Linear（）的相似函数是什么？任何替换或类似的功能，以获得相同的结果。

Answer 1

只需在slim.fully_connected中使用tf.contrib.slim。

在火炬中，

module= nn.Linear(input_dims, output_dims)

等同于

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。