tensorflow中custom_getter的概念及使用

这是一种在创建变量时,潜在地重写某些内容的方法。在处理某些特定的问题时非常方便。

从概念上讲,custom_getter类似于python修饰符:定义一个这样的函数,它可以获取原函数及其参数作为它的参数,但必须返回结果而不是经修饰后的函数。

在tf.variable_scope()中,你也可以指定custom_getter作为一个参数,这样你可以一次传入多个变量。

下面给出一个例子。

假设你有大量的代码来创建某种类型的网络。一天早上醒来,你想试试L2规范化所有变量是否有助于提高网络性能。不必重新改你所创建的网络层及其内部变量的代码,可以直接将整个网络封闭在一个scope并执行以下操作:

with tf.variable_scope( "L2", custom_getter =
    lambda getter, name, shape, *args, **kwargs:
        tf.nn.l2_normalize( getter( name = name, shape = shape, *args, **kwargs ) ) ):
    # the original network here

这将自动地L2规范化网络中的所有变量。当然,如果你不想对所有的变量都这样做,你可以编写更多的代码来筛选变量。

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值