Function ste_backward 

pub fn ste_backward(grad_output: &Tensor) -> Tensor

Compute STE backward pass (identity gradient).

In the backward pass, gradients flow through unchanged. This is handled automatically by the tensor operations, but this function is provided for clarity.

Arguments

• grad_output - Gradient from the next layer

Returns

The same gradient (identity function)