o i@s$ddlmmZGdddZdS)Nc@s eZdZddZeddZdS) NamedTensorcCst|tjjjtjjjBstdt||jdurt dt |jt |jkr9t dt |jdt |j||_ ||_ dS)Nz2Value must be a Tensor/ConvertibleTensor, but got zValue must have a shapezShape mismatch: z vs ) isinstancetracer signature classicalTensorConvertibleTensor TypeErrortypeshape ValueErrortuplevalueexpr)selfrrrU/home/ubuntu/.local/lib/python3.10/site-packages/einx/_src/namedtensor/namedtensor.py__init__s   zNamedTensor.__init__cCs t|jjS)N)lenrr )rrrrndims zNamedTensor.ndimN)__name__ __module__ __qualname__rpropertyrrrrrrs r)einx._src.tracer_srcrrrrrrs