如何使用TensorFlow自定义layer并有效集成至计算图中？

本文共计345个文字，预计阅读时间需要2分钟。

目标+使用用户自定义的layer结合TensorFlow自带的layer组成多层layer的算子图。实现功能+包括2D图像进行滑动窗口平移，并通过自定义的操作layer返回结果。+import tensorflow as tf+import numpy as np sess=

目的

将用户自定义的layer结合tensorflow自带的layer组成多层layer的计算图。

实现功能

对2D图像进行滑动窗口平均，并通过自定义的操作layer返回结果。

import tensorflow as tf import numpy as np sess = tf.Session() #将size设为[1, 4, 4, 1]是因为tf中图像函数是处理四维图片的。