tfa.image.utils.wrap

View source on GitHub

Returns image with an extra channel set to all 1s.