TensorFlow函数教程:tf.nn.dropout

本文深入解析了tf.nn.dropout函数的用法与原理,该函数在神经网络中用于实现dropout正则化,通过设定保留概率来随机关闭部分神经元,防止过拟合。文章详细介绍了函数的参数设置,包括保留概率、噪声形状等,并解释了如何调整这些参数以优化模型表现。

tf.nn.dropout函数

tf.nn.dropout(
    x,
    keep_prob,
    noise_shape=None,
    seed=None,
    name=None
)

定义在:tensorflow/python/ops/nn_ops.py.

请参阅指南:层(contrib)>用于构建神经网络层的高级操作,神经网络>激活函数

该函数用于计算dropout.

使用概率keep_prob,输出按照1/keep_prob的比例放大输入元素,否则输出0.缩放是为了使预期的总和不变.

默认情况下,每个元素都是独立保留或删除的.如果已指定noise_shape,则必须将其广播为x的形状,并且只有具有noise_shape[i] == shape(x)[i]的维度才作出独立决定.

例如,如果shape(x) = [k, l, m, n]并且noise_shape = [k, 1, 1, n],则每个批处理和通道组件将独立保存,并且每个行和列将保留或不保留在一起.

参数:

  • x:一个浮点型Tensor.
  • keep_prob:一个标量Tensor,它与x具有相同类型.保留每个元素的概率.
  • noise_shape:类型为int32的1维Tensor,表示随机产生的保持/丢弃标志的形状.
  • seed:一个Python整数.用于创建随机种子.
  • name:此操作的名称(可选).

返回:

该函数返回与x具有相同形状的Tensor.

可能引发的异常:

  • ValueError:如果keep_prob不在(0, 1]或如果x不是浮点型Tensor.

转载于:https://www.cnblogs.com/tsdblogs/p/10405268.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值