dropout
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
x |
|
x |
|
√ |
|
√ |
|
x |
针对
功能说明
和tf.nn.dropout功能相同。将输入Tensor按照1/keep_prob缩放，并且输入Tensor保留的概率为keep_prob，否则输出0，并且最终输出Tensor的shape和输入Tensor的shape保持一致。
函数原型
1 | 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() |