Tensorflow 2:如何配置自定义层的速度

2024-07-02 04:45:17 发布

您现在位置:Python中文网/ 问答频道 /正文

非常简单和直接的问题:什么是加速测试自定义keras层的最简单方法?为了澄清,我正在对keras层进行子类化,并在tensorflow ops中实现我的逻辑。由于它广泛使用稀疏张量,我有点担心速度。在标准python中,我可能会使用类似timeit的东西,但不确定这是否适用于tensorflow(可能适用于急切执行)?同时测试cpu和gpu也很好


Tags: 方法标准gputensorflow逻辑cpu速度ops