PatchEmbedding constructor

PatchEmbedding({
  1. required int patchSize,
  2. required int inChannels,
  3. required int embedDim,
})

Implementation

PatchEmbedding({
  required this.patchSize,
  required this.inChannels,
  required this.embedDim,
})  : projection =
          Layer.fromNeurons(patchSize * patchSize * inChannels, embedDim),
      norm = LayerNorm(embedDim);