多输入问题

你好,如果我想使用两个输入来分割图像,我应该如何修改代码?例如:RGB:(B,3,H,W) 深度:(B,1,H,W) 我使用RGB和深度作为输入,最后会得到分割图。所以我想看到网络中某些层的CAM图。