83797400128d41910d87e957131f29fd466f4777,example/extensions/lib_custom_op/test_transposecsr.py,,,#,30

Before Change

print("Compute Results:")
print("data:", out[0].data.asnumpy())
print("indices:", out[0].indices.asnumpy())
print("indptr:", out[].indptr.asnumpy())

out2 = exe2.forward()
out2 = exe2.forward()

After Change

f = mx.sym.my_state_transposecsr(d, test_kw=200)

block = mx.gluon.nn.SymbolBlock(e, [d])
out = block(a)
print("Compute Results:")
print("data:", out.data.asnumpy())
print("indices:", out.indices.asnumpy())
print("indptr:", out.indptr.asnumpy())

Instances