文档
注册

dropout

功能说明

和tf.nn.dropout功能相同。将输入Tensor按照1/keep_prob缩放,并且输入Tensor保留的概率为keep_prob,否则输出0,并且最终输出Tensor的shape和输入Tensor的shape保持一致。

函数原型

def dropout(x, keep_prob, noise_shape=None, seed=None, name=None)

参数说明

参数名

输入/输出

描述

x

输入

输入Tensor,float类型。

keep_prob

输入

标量Tensor,float类型。代表每个元素保留的概率。

noise_shape

输入

一维Tensor,int32类型。表示随机生成的keep/drop标志的形状。

seed

输入

随机数种子。

name

输入

网络层的名称。

返回值

tensor:对输入x执行完dropout操作之后的输出Tensor。

调用示例

1
2
from npu_bridge.npu_init import *
layers = npu_ops.dropout()
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词