Scaled Token Embedding LookupDeep-ML | Problem