Tensorflow C ++ API没有dropout()
函数。哪个功能可以取代这个dropout()
功能?
您始终可以从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
获得具有正确形状的随机张量。其余的应该是微不足道的。