Skip to main content

embed_diagonal

tenferro_tensor::cpu::structural

Function embed_diagonal 

Source 
pub fn embed_diagonal(
    input: &Tensor,
    axis_a: usize,
    axis_b: usize,
) -> Result<Tensor>