tfe_svd_rrule_f64

Function tfe_svd_rrule_f64 

Source
#[unsafe(no_mangle)]
pub unsafe extern "C" fn tfe_svd_rrule_f64( tensor: *const TfeTensorF64, left: *const usize, left_len: usize, right: *const usize, right_len: usize, max_rank: usize, cutoff: f64, cotangent_u: *const TfeTensorF64, cotangent_s: *const TfeTensorF64, cotangent_vt: *const TfeTensorF64, status: *mut tfe_status_t, ) -> *mut TfeTensorF64
Expand description

Reverse-mode rule (VJP) for SVD.

§Safety

  • tensor must be valid and non-null.
  • left and right must point to valid index arrays.
  • status must be valid.

§Examples (C)

tfe_tensor_f64 *grad = tfe_svd_rrule_f64(a, left, 1, right, 1, 0, -1.0, NULL, cot_s, NULL, &status);