C ++没有dp()函数

时间:2018-08-23 06:58:03

标签: c++ tensorflow

Tensorflow C ++ API没有dropout()函数。哪个函数可以代替此dropout()函数?

1 个答案:

答案 0 :(得分:1)

您始终可以从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

使用tensorflow::ops::RandomUniform获得形状正确的随机张量。其余的应该是微不足道的。