maxframe 2.0.0b2__cp311-cp311-win32.whl → 2.2.0__cp311-cp311-win32.whl

maxframe/tensor/indexing/fill_diagonal.py

@@ -18,7 +18,7 @@ import numpy as np
 
 from ... import opcodes
 from ...core import ENTITY_TYPE, EntityData
-from ...serialization.serializables import AnyField, BoolField, Int32Field, KeyField
+from ...serialization.serializables import AnyField, BoolField, Int32Field
 from ..core import Tensor
 from ..datasource import tensor as astensor
 from ..operators import TensorOperator, TensorOperatorMixin
@@ -27,20 +27,14 @@ from ..operators import TensorOperator, TensorOperatorMixin
 class TensorFillDiagonal(TensorOperator, TensorOperatorMixin):
     _op_type_ = opcodes.FILL_DIAGONAL
 
-    _input = KeyField("input")
     val = AnyField("val", default=None)
     wrap = BoolField("wrap", default=None)
     # used for chunk
     k = Int32Field("k", default=None)
 
-    @property
-    def input(self):
-        return self._input
-
     @classmethod
     def _set_inputs(cls, op: "TensorFillDiagonal", inputs: List[EntityData]):
         super()._set_inputs(op, inputs)
-        op._input = op._inputs[0]
         if len(op._inputs) == 2:
             op.val = op._inputs[1]
 

maxframe/tensor/linalg/__init__.py

@@ -12,16 +12,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from .cholesky import TensorCholesky, cholesky
 from .dot import TensorDot, dot
+from .einsum import TensorEinsum, einsum
 from .inner import inner, innerproduct
 from .inv import TensorInv, inv
+from .lstsq import lstsq
 from .lu import TensorLU, lu
 from .matmul import TensorMatmul, matmul
+from .matrix_norm import matrix_norm
+from .norm import TensorNorm, norm
 from .qr import TensorQR, qr
+from .solve import solve
 from .solve_triangular import TensorSolveTriangular, solve_triangular
 from .svd import TensorSVD, svd
 from .tensordot import TensorTensorDot, tensordot
 from .vdot import vdot
+from .vector_norm import vector_norm
 
 
 def _install():