Function dfdx::tensor_ops::lower_tri
source · pub fn lower_tri<S: Shape, E: Dtype, D: TriangleTensor<E>, T: Tape<E, D>>(
t: Tensor<S, E, D, T>,
diagonal: impl Into<Option<isize>>
) -> Tensor<S, E, D, T>where
Tensor<S, E, D, T>: TryMul<Tensor<S, E, D>> + HasErr<Err = D::Err>,
Expand description
Applies a 2D lower triangular mask by setting values above the diagonal to E::default()
.