Skip to main content

extract_diagonal

tenferro_cpu

Function extract_diagonal 

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