pub fn stop_gradient(a: impl AsRef<Array>) -> Result<Array>
Stop gradients from being computed.
The operation is the identity but it prevents gradients from flowing through the array.