【tf】tf.nn.dropout

dropout是神经网络中用来防止过拟合的技巧,在每一轮训练时随机丢弃一些神经元。

tf.nn.dropout(x,
              keep_prob,
              noise_shape=None,
              seed=None,
              name=None)
  • x:一个浮点型Tensor,注意不能是sparsetensor。
  • keep_prob:一个标量Tensor,取值(0,1]间,表示每个元素的保留概率,每一个参数都将按这个比例随机变更。
  • noise_shape:int32的一维Tensor,表示随机产生的保持/丢弃标志的形状。
  • seed:一个Python整数,用于创建随机种子。
  • name:此操作的名称(可选)。
x = tf.Variable([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9],
                 [10, 11, 12]], dtype=tf.float32)
keep_prob = 0.5
a = tf.nn.dropout(x,
                  keep_prob)
print(a)
>>> tf.Tensor([[ 2.  4.  6.]
			   [ 8.  0. 12.]
			   [ 0.  0. 18.]
			   [ 0. 22. 24.]], shape=(4, 3), dtype=float32)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值