feat: add functional pad on segmentation mask#5866

Merged

vfdev-5 merged 6 commits intopytorch:mainfrom federicopozzi33:feat/5782-proto-mask-pad

May 2, 2022

+69

Commits on Apr 28, 2022

feat: add functional pad on segmentation mask
Federico Pozzi
committed
test: add basic correctness test with random masks
Federico Pozzi
committed
test: add all padding options
Federico Pozzi
committed
fix: pr comments
Federico Pozzi
committed
fix: tests
Federico Pozzi
committed
refactor: reshape tensor in 4d, then pad
Federico Pozzi
committed