tfe_tensor_f64_shape

Function tfe_tensor_f64_shape 

Source
#[unsafe(no_mangle)]
pub unsafe extern "C" fn tfe_tensor_f64_shape( tensor: *const TfeTensorF64, out_shape: *mut usize, status: *mut tfe_status_t, )
Expand description

Write the shape of the tensor into the caller-provided buffer.

The caller must allocate out_shape with at least tfe_tensor_f64_ndim(tensor) elements.

ยงSafety

  • tensor must be a valid, non-null tensor pointer.
  • out_shape must point to a buffer with at least tfe_tensor_f64_ndim(tensor) elements.
  • status must be a valid, non-null pointer.