类似于 Lua nn.Linear() 的 tensorflow 函数

我试图将一个lua代码实现到tensorflow中。在tensorflow中有哪些与nn.Linear()类似的函数?有没有替代函数或类似函数能够得到同样的结果。

点赞
用户7045779
用户7045779

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

在Torch中,

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)

在tensorflow中,不考虑数据的批量操作。

请查看 tf.contrib.slim documentation

2017-02-05 21:59:05