tensorflo实现Droppath
2022/4/16 23:43:10
本文主要是介绍tensorflo实现Droppath,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
# def drop_path(inputs, keep_prob, is_training=True, scope=None): def drop_path(inputs, keep_prob, is_training=True): """Drops out a whole example hiddenstate with the specified probability. """ # with tf.name_scope(scope, 'drop_path', [inputs]): net = inputs if is_training: batch_size = tf.shape(net)[0] noise_shape = [batch_size, 1, 1, 1] random_tensor = keep_prob random_tensor += tf.random_uniform(noise_shape, dtype=tf.float32) binary_tensor = tf.floor(random_tensor) net = tf.div(net, keep_prob) * binary_tensor return net class DropPath(keras.layers.Layer): def __init__(self, drop_prob=None): super(DropPath, self).__init__() self.drop_prob = drop_prob def call(self, inputs,training=None): return drop_path(inputs, self.drop_prob, training)
搜索
复制
这篇关于tensorflo实现Droppath的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-13TiDB + ES:转转业财系统亿级数据存储优化实践
- 2024-05-09“2024鸿蒙零基础快速实战-仿抖音App开发(ArkTS版)”实战课程已上线
- 2024-05-09聊聊如何通过arthas-tunnel-server来远程管理所有需要arthas监控的应用
- 2024-05-09log4j2这么配就对了
- 2024-05-09nginx修改Content-Type
- 2024-05-09Redis多数据源,看这篇就够了
- 2024-05-09Google Chrome驱动程序 124.0.6367.62(正式版本)去哪下载?
- 2024-05-09有没有大佬知道这种数据应该怎么抓取呀?
- 2024-05-09这种运行结果里的10.100000001,怎么能最快改成10.1?
- 2024-05-09企业src漏洞挖掘-有意思的命令执行