PatchMerging constructor

PatchMerging(
  1. int dim
)

Implementation

PatchMerging(this.dim) {
  norm = LayerNorm(4 * dim);
  reduction = Linear(4 * dim, 2 * dim, useBias: false);
}