Struct OutputProjection 

pub struct OutputProjection { /* private fields */ }

Output projection to predict next byte distribution

Implementations

impl OutputProjection

pub fn new(embed_dim: usize, rng: &mut StdRng) -> Self

pub fn set_temperature(&mut self, temp: f32)

pub fn forward(&self, hidden: &[f32]) -> Vec<f32>

pub fn sample(&self, probs: &[f32], rng: &mut StdRng) -> u8

pub fn argmax(&self, probs: &[f32]) -> u8

Trait Implementations

impl Clone for OutputProjection

fn clone(&self) -> OutputProjection

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OutputProjection

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for OutputProjection

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for OutputProjection

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.