modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D#
- class modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D[source]#
Bases:
torch.nn.Module
An upsampling layer with an optional convolution.
- Parameters
channels – channels in the inputs and outputs.
use_conv – a bool determining if a convolution is applied.
use_conv_transpose –
out_channels –
Methods
forward
- __call__(*args: Any, **kwargs: Any) Any #
Call self as a function.
- __init__(channels, use_conv=False, use_conv_transpose=False, out_channels=None, name='conv')[source]#
- static __new__(cls, *args: Any, **kwargs: Any) Any #