tensorflow 写模块

请问就是这么一个注意力模块，用tensorflow怎么写出来啊，俺没找到源码，两个卷积的卷积核都是3*3，第一个输出通道是256，第二个输出通道是2

img

tensorflow 2.0 实现卷积注意力模块
https://blog.csdn.net/highlevels/article/details/112302884