FixedLenFeature

背景

在使用dataset进行样本解析之前,我们需要先定义一个解析字典,告诉dataset如何去解析每个样本,这个字典就是用来指定对于每条输入样本的每一列应该用什么的feature去解析,dataset默认提供了FixedLenFeature,VarLenFeature,FixedLenSequenceFeature等。顾名思义,FixedLenFeature用于处理定长的特征,VarLenFeature用于处理变长的特征,FixedLenSequenceFeature用于处理定长的序特征。

案例

features_config[fea_name] = tf.FixedLenFeature([], tf.int64)
features_config[fea_name] = tf.FixedLenFeature([1], tf.int64)

FixedLenFeature在处理特征时,会根据输入的shape来得到相应的输出tensor的shape。
当输入shape = []时,输出tensor的shape=(batch_size,),
当输入shape=[k]时,输出tensor的shape= (batch_size,k)

tf在coding的过程中很多时候会因为输入tensor的shape问题而导致执行失败。想起当时在看Andrew NG的教学视频时,他也提到了这个问题,他给建议是,每次在进行计算之前一定要对tensor进行reshape从而保证后续的操作能正常进行。

原文链接:https://www.jianshu.com/p/581d89590829

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值