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