Bind compute dims
Feature:
-
Allow to define the computation of output dimension using a python function
Usage:
import aidge_core
w0 = aidge_core.Producer([25, 25], name="W0")
matmul0 = aidge_core.GenericOperator("MatMul", 1, 1, 1, name="MatMul0")
graph_view = aidge_core.sequential([w0, matmul0])
def myComputationMethod(a):
print(a)
return a
matmul0.get_operator().set_compute_output_dims(myComputationMethod)
matmul0.get_operator().compute_output_dims()
-
Add identity
function to the binding. usefull ? (equivalent tolambda x:x```which is shorter than
aidge_core.GenericOperatorOp.identity``...) -
Add unittest
Edited by Cyril Moineau
Merge request reports
Activity
Filter activity
changed milestone to %v0.1.0
requested review from @vtemplier and @olivierbichler
assigned to @cmoineau
added 1 commit
- d84a69c7 - [GenericOperator] Add set_compute_output_dims & compute_output_dims method + associated unittest.
mentioned in commit a4d0063f
Please register or sign in to reply