tfe_svd_frule_f64

Function tfe_svd_frule_f64 

Source
#[unsafe(no_mangle)]
pub unsafe extern "C" fn tfe_svd_frule_f64( tensor: *const TfeTensorF64, left: *const usize, left_len: usize, right: *const usize, right_len: usize, max_rank: usize, cutoff: f64, tangent: *const TfeTensorF64, u_out: *mut *mut TfeTensorF64, s_out: *mut *mut TfeTensorF64, vt_out: *mut *mut TfeTensorF64, status: *mut tfe_status_t, )
Expand description

Forward-mode rule (JVP) for SVD.

§Safety

  • tensor must be valid and non-null.
  • left and right must point to valid index arrays.
  • u_out, s_out, vt_out must be valid, non-null pointers.
  • status must be valid.

§Examples (C)

tfe_svd_frule_f64(a, left, 1, right, 1, 0, -1.0, da, &du, &ds, &dvt, &status);