简体   繁体   中英

C++ did not have the dp() function

Tensorflow C++ API did not have the dropout() function. Which function can replace this dropout() function ?

You can always copy the algorithm from python ( link ).

random_tensor = keep_prob
random_tensor += random_ops.random_uniform(
    noise_shape, seed=seed, dtype=x.dtype)
# 0. if [keep_prob, 1.0) and 1. if [1.0, 1.0 + keep_prob)
binary_tensor = math_ops.floor(random_tensor)
ret = math_ops.div(x, keep_prob) * binary_tensor

Use tensorflow::ops::RandomUniform to get the random tensor with the right shape. The rest should be trivial.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM