This could allow for integration into, for example, some type of U-net segmentation network. Remember to test with gradcheck.