I would like to add channels to a tensor, so that from [N,C,H,W] it becomes [N, C+rpad, H, W], potentially filled with zeros for the new channels, all that in a
forward() function of a
From what I gathered, padding in the width and height dimensions is implemeted in
F.pad(), which calls functions from
nn._functions.padding, such as the ones of
ConstantPad2d. Unfortunately padding channels is missing here.
I also see that a
legacy module implement some of those things, but it does not seem to create new
Variables, and I am not sure if I should use something like this.
Anyways, I am open to suggestions as to clean&concise ways to implement zero-padding in the channels to use in a