modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Downsample2D#

class modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Downsample2D[source]#

Bases: torch.nn.Module

A downsampling layer with an optional convolution.

Parameters

  • channels – channels in the inputs and outputs.

  • use_conv – a bool determining if a convolution is applied.

  • out_channels –

  • padding –

Methods

forward

__call__(*args: Any, **kwargs: Any) → Any#

Call self as a function.

__init__(channels, use_conv=False, out_channels=None, padding=1, name='conv')[source]#
static __new__(cls, *args: Any, **kwargs: Any) → Any#

previous

modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D

next

modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.ResnetBlock2D