发新帖

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

[复制链接]
644 3

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
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))
我知道答案 回答被采纳将会获得10 金币 已有3人回答
本楼点评(0) 收起

精彩评论3

AirLRJ  TF荚荚  发表于 2018-4-17 09:32:15 | 显示全部楼层
您好,咱们可以直接来看代码:

tf.layers.dense 会创建一个 tf.layers.Dense 的object

在inputs 的维度大于2的时候,会调用一个broadcast的点乘,其他时候则是调用一个矩阵相乘
本楼点评(0) 收起
TianLin  TF荚荚  发表于 2018-4-17 09:43:00 | 显示全部楼层
因为数据输入都是一组mini batch的数据点,第一维是batch_size,第二维代表每个数据点,[batch_size, input_size]
本楼点评(0) 收起
ViolinSolo  TF豆豆  发表于 2018-7-3 20:07:28 | 显示全部楼层
因为默认是带batch的,,你直接写一维,怕是keras写多了
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

5

积分
快速回复 返回顶部 返回列表