EmbeddingParams

data class EmbeddingParams(val numEmbeddings: Int, val embeddingDim: Int, val paddingIdx: Int? = null, val maxNorm: Float? = null, val scaleGradByFreq: Boolean = false)(source)

Parameters for configuring an Embedding layer.

This is a lightweight holder used to describe the table size and optional behaviors. Some fields are placeholders for future features (no-ops for now) to keep parity with common NN APIs.

Constructors

Link copied to clipboard
constructor(numEmbeddings: Int, embeddingDim: Int, paddingIdx: Int? = null, maxNorm: Float? = null, scaleGradByFreq: Boolean = false)

Properties

Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard