Function tfe_einsum_frule_f64 

#[unsafe(no_mangle)]
pub unsafe extern "C" fn tfe_einsum_frule_f64(
    _subscripts: *const c_char,
    _primals: *const *const TfeTensorF64,
    _num_operands: usize,
    _tangents: *const *const TfeTensorF64,
    _status: *mut tfe_status_t,
) -> *mut TfeTensorF64

Forward-mode rule (JVP) for einsum.

Returns the output tangent. Elements of tangents may be null (interpreted as zero tangent for that operand).

Safety

• subscripts must be a valid null-terminated C string.
• primals must point to an array of num_operands valid tensor pointers.
• tangents must point to an array of num_operands tensor pointers (elements may be null).
• status must be a valid, non-null pointer.

Examples (C)

const tfe_tensor_f64 *primals[] = {a, b};
const tfe_tensor_f64 *tangents[] = {da, NULL};  // no tangent for b
tfe_status_t status;
tfe_tensor_f64 *dc = tfe_einsum_frule_f64(
    "ij,jk->ik", primals, 2, tangents, &status);
tfe_tensor_f64_release(dc);