tf.layers.dense 输入 tensor 至少是 2 维?

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