fix: respect channel_axis in pad_to_size for channels-first images - #112

fix: respect channel_axis in pad_to_size for channels-first images#112

copybara-service[bot] merged 2 commits into

google-deepmind:mastergoogle-deepmind/dm_pix:masterfrom

Nas01010101:fix/pad-to-size-channels-firstNas01010101/dm_pix:fix/pad-to-size-channels-firstCopy head branch name to clipboard