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