LanguageModelOutput

Value parameters

encoded

encoded: float tensor of size (batch, sequence length, embedding dimension) holds per token embeddings

languageModelLogits

float tensor of size (batch, sequence length, vocabulary size) holds per token logits. Use logSoftMax(dim=2) to get log probabilities.

Attributes

Graph

Reset zoomHide graphShow graph

Supertypes

trait Serializable

trait Product

trait Equals

class Object

trait Matchable

class Any

Show all