dropout的实现

2020-09-23  本文已影响0人  康英永

tensorflow关于dropout的实现如下:

ret = math_ops.div(x, keep_prob) * binary_tensor

其中:

x为前置网络得出的张量,keep_prob=1-dropout,binary_tensor为保留的元素的标识,1为保留,0为drop,shape与x一致。

上一篇 下一篇

猜你喜欢

热点阅读