Getting Started
Core Workflows
Search & Diagnostics
API Reference
Functions
instance_norm
layer_norm
Classes
DownsampleBlock
LayerNorm
LayerNorm that supports two data formats: channels_last (default) or channels_first.
MinPool2d
ResBlockDecBasic
ResBlockDecBottleneck
ResBlockEncBasic
ResBlockEncBottleneck
ResizeConv2d
UpsampleBlock
vanilla_Decoder
vanilla_Encoder