PatchMerge
Patch merging layer from Swin-Transformer (opens in a new tab).
PatchMerge(dim=96, expansion_factor=2.0)Forward
(x: jaxtyping.Float[Tensor, '... n d_in']) -> jaxtyping.Float[Tensor, '... n/4 d_out']Patch merging layer from Swin-Transformer (opens in a new tab).
PatchMerge(dim=96, expansion_factor=2.0)(x: jaxtyping.Float[Tensor, '... n d_in']) -> jaxtyping.Float[Tensor, '... n/4 d_out']