tfa.image.utils.wrap

View source on GitHub

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