Function triu_device

pub fn triu_device(
    a: impl AsRef<Array>,
    k: impl Into<Option<i32>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>

Zeros the array below the given diagonal

Params

• a: input array
• k: diagonal of the 2D array. Default to 0