Correct way to use the embedding layer

[VarLad]

Hello, I was wondering whats the correct way to use the Embedding layer.
Looking at the docs, I tried:

using Lux
x = Embedding(10 => 10)
y = x([1,2,3])
z = x(5) 

but that doesn't work.

[avik-pal]

That is not the correct way to call Lux layers. See the manual on the interface specification.

x = Embedding(10 => 10)
ps, st = Lux.setup(Random.default_rng(), x)

y = x([1,2,3], ps, st)
z = x(5, ps, st)