Struct Embedding 

pub struct Embedding { /* private fields */ }

Embedding layer

Implementations

impl Embedding

pub fn new(input_dim: usize, output_dim: usize) -> Self

Create new Embedding layer

pub fn mask_zero(self, mask_zero: bool) -> Self

Set mask zero

pub fn name(self, name: &str) -> Self

Set layer name

Trait Implementations

impl KerasLayer for Embedding

fn call(&self, input: &ArrayD<f64>) -> Result<ArrayD<f64>>

Forward pass through the layer

fn build(&mut self, _input_shape: &[usize]) -> Result<()>

Build the layer (called during model compilation)

fn compute_output_shape(&self, input_shape: &[usize]) -> Vec<usize>

Compute output shape given input shape

fn count_params(&self) -> usize

Get number of parameters

fn get_weights(&self) -> Vec<ArrayD<f64>>

Get trainable parameters

fn set_weights(&mut self, weights: Vec<ArrayD<f64>>) -> Result<()>

Set trainable parameters

fn built(&self) -> bool

Check if layer is built

fn name(&self) -> &str

Get layer name