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);