Image tensor spliting

You might need to adapt the padding for your input based on this post and make sure the output shape is as expected.