The nvmath.bindings module is experimental and potentially subject to future changes.

estimate_workspace_size

nvmath.bindings.cutensor.estimate_workspace_size(

intptr_t handle,

intptr_t desc,

intptr_t plan_pref,

int workspace_pref,

) → uint64_t

See cutensorEstimateWorkspaceSize.