pyvene.models.basic_utils.embed_to_distrib

pyvene.models.basic_utils.embed_to_distrib#

embed_to_distrib(model, embed, log=False, logits=False)[source]#

Convert an embedding to a distribution over the vocabulary