Function init_class_emb 

pub(crate) fn init_class_emb(
    n: usize,
    width: usize,
    device: &Device,
) -> Option<Param<Tensor<2>>>

Build the embedding param for n class markers of the given width (None when there are none — Burn has no zero-width tensors).