tensorflow 替换网络层

pytorch_grad_cam 翻译成tensorflow代码时。需要将tensorflow的某层网络根据名称或类型，更改为新的网络层。
请问怎么替换网络层？