tfe_einsum_f64

Function tfe_einsum_f64 

Source
#[unsafe(no_mangle)]
pub unsafe extern "C" fn tfe_einsum_f64( _subscripts: *const c_char, _operands: *const *const TfeTensorF64, _num_operands: usize, _status: *mut tfe_status_t, ) -> *mut TfeTensorF64
Expand description

Execute einsum using string notation.

Returns a new tensor. The caller must release it with tfe_tensor_f64_release.

§Safety

  • subscripts must be a valid null-terminated C string.
  • operands must point to an array of num_operands valid tensor pointers.
  • status must be a valid, non-null pointer.

§Examples (C)

const tfe_tensor_f64 *ops[] = {a, b};
tfe_status_t status;
tfe_tensor_f64 *c = tfe_einsum_f64("ij,jk->ik", ops, 2, &status);
assert(status == TFE_SUCCESS);
tfe_tensor_f64_release(c);