tf.layers.dense 输入 tensor 至少是 2 维吗??我输入的 1 维一直报错:
self.s = tf.placeholder (tf.float64, [None]) l_eval = tf.layers.dense (self.s, 10, tf.nn.relu, kernel_initializer=tf.random_normal_initializer (0, 0.1))
提问人:Y.Z.kun,发帖时间:2018-4-16 21:19:33
tf.layers.dense 输入 tensor 至少是 2 维吗??我输入的 1 维一直报错:
self.s = tf.placeholder (tf.float64, [None]) l_eval = tf.layers.dense (self.s, 10, tf.nn.relu, kernel_initializer=tf.random_normal_initializer (0, 0.1))
提问人:Y.Z.kun,发帖时间:2018-4-16 21:19:33
您好,咱们可以直接来看代码:
tf.layers.dense 会创建一个 tf.layers.Dense 的 object
在 inputs 的 维度大于 2 的时候,会调用一个 broadcast 的点乘,其他时候则是调用一个 矩阵相乘。
AirLRJ,发表于 2018-4-17 09:32:15
因为数据输入都是一组 mini batch 的数据点,第一维是 batch_size,第二维代表每个数据点,[batch_size, input_size]
TianLin,发表于 2018-4-17 09:43:00
因为默认是带 batch 的,,你直接写一维,怕是 keras 写多了
ViolinSolo,发表于 2018-7-3 20:07:28