C ++没有dp()函数

问题描述 投票:1回答:1

Tensorflow C ++ API没有dropout()函数。哪个功能可以取代这个dropout()功能?

c++ tensorflow
1个回答
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获得具有正确形状的随机张量。其余的应该是微不足道的。

© www.soinside.com 2019 - 2024. All rights reserved.