Tensorflow C ++ API没有dropout()
函数。哪个函数可以代替此dropout()
函数?
答案 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
获得形状正确的随机张量。其余的应该是微不足道的。