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

block_sparse_contract

nvmath.bindings.cutensor.block_sparse_contract(

intptr_t handle,

intptr_t plan,

intptr_t alpha,

a,

b,

intptr_t beta,

c,

d,

intptr_t workspace,

uint64_t workspace_size,

intptr_t stream,

)

See cutensorBlockSparseContract.