dataset based ONNX implementation for vector sampling.

Input

3722-dim embedding

Encoder

10 x MLP with 64 heads

Output

f1 projection