class

MicroGPT::BigramTable

Inherits Reference < Object

Constructors

new(data : Array(Int32), vocab_size : Int32)
Source

Instance methods

lookup(input_ids : Array(Int32)) : Mat

Lookup: returns [seq_len, vocab_size] matrix of bigram distributions

Source
table
Source
vocab_size
Source