tensorflow
天才da熊猫
这个作者很懒,什么都没留下…
展开
-
tf算子计算流程统计
tf.depth_to_space()import numpy as npimport tensorflow.compat.v1 as tftf.disable_v2_behavior()if __name__ == "__main__": block_size = 3 i1 = tf.linspace(1.0,288.0,288) i2 = tf.reshape(i1,[1,4,4,18]) i3 = tf.depth_to_space(i2,block_size=block_si.原创 2020-10-15 16:20:30 · 271 阅读 · 0 评论 -
python broadcast机制的模拟实现
tensorflow的算术操作:mul/add/sub等op都支持broadcast机制,该机制支持不同维度的计算,但是在对维度进行逆向比较时需要满足以下要求:1)二者维度相同2)二者维度有一个为13)如果维度大小不一致,需要用1来对维度小的数据进行扩展,在进行上述判断;如:a:[256,256,3]、b:[3]这样的维度,需要先将b扩展至与a一致,将b扩展至[1,1,3],再对a...原创 2019-06-27 09:50:17 · 259 阅读 · 0 评论 -
tensflow SpaceToBatchND + conv + BatchToSpaceND
参考博客1、https://blog.csdn.net/murdock_c/article/details/874702482、https://blog.csdn.net/silence2015/article/details/79748729总结一下:以一个1x6x6x1(nhwc)为例,假如卷积核大小3x3,stride = 1,dilation=2的卷积。输入kernel...原创 2019-07-19 14:56:20 · 1084 阅读 · 0 评论 -
tensorflow量化部分整理ing
主要量化要求https://github.com/tensorflow/tensorflow/blob/a290c288bd05b76b22b273797833749018657e30/tensorflow/lite/g3doc/performance/quantization_spec.md1、卷积的量化参数(权重、输出的min max由fakequant给定)2、Pooling、Res...原创 2019-08-14 15:36:21 · 1026 阅读 · 0 评论 -
tensorflow resize代码解析
tensorflow resizebilinear函数为例,其scale的计算与resize函数的align_corner有关:inline float CalculateResizeScale(int64 in_size, int64 out_size, bool align_corners) { return (al...原创 2019-09-25 14:08:41 · 3248 阅读 · 1 评论