maxframe 1.3.1__cp310-cp310-macosx_10_9_universal2.whl → 2.0.0b1__cp310-cp310-macosx_10_9_universal2.whl

maxframe/tensor/arithmetic/multiply.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -14,12 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import functools
+
 import numpy as np
 
 from ... import opcodes
+from ...serialization.serializables import BoolField
+from ..datasource import scalar
 from ..utils import infer_dtype
-from .core import TensorBinOp
-from .utils import arithmetic_operator
+from .core import TensorBinOp, TensorMultiOp
+from .utils import TreeReductionBuilder, arithmetic_operator
 
 
 @arithmetic_operator(sparse_mode="binary_or")
@@ -79,3 +81,34 @@ def multiply(x1, x2, out=None, where=None, **kwargs):
 def rmultiply(x1, x2, **kwargs):
     op = TensorMultiply(**kwargs)
     return op.rcall(x1, x2)
+
+
+class TensorTreeMultiply(TensorMultiOp):
+    _op_type_ = opcodes.TREE_MULTIPLY
+    _func_name = "multiply"
+
+    ignore_empty_input = BoolField("ignore_empty_input", default=False)
+
+    def __init__(self, sparse=False, **kw):
+        super().__init__(sparse=sparse, **kw)
+
+    @classmethod
+    def _is_sparse(cls, *args):
+        if not args or all(np.isscalar(x) for x in args):
+            return False
+        if all(
+            np.isscalar(x) or (hasattr(x, "issparse") and x.issparse()) for x in args
+        ):
+            return True
+        return False
+
+
+@infer_dtype(lambda *args: functools.reduce(np.multiply, args))
+def tree_multiply(*args, combine_size=None, **kwargs):
+    class MultiplyBuilder(TreeReductionBuilder):
+        def _build_reduction(self, inputs, final=False):
+            op = TensorTreeMultiply(args=inputs, **kwargs)
+            return op(*inputs)
+
+    args = [scalar(a) if np.isscalar(a) else a for a in args]
+    return MultiplyBuilder(combine_size).build(args)

maxframe/tensor/arithmetic/nan_to_num.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/negative.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/nextafter.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/not_equal.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/positive.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/power.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/rad2deg.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/radians.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/real.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/reciprocal.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -22,11 +20,15 @@ from .core import TensorUnaryOp
 from .utils import arithmetic_operator
 
 
-@arithmetic_operator(sparse_mode="unary")
+@arithmetic_operator(sparse_mode="always_false")
 class TensorReciprocal(TensorUnaryOp):
     _op_type_ = opcodes.RECIPROCAL
     _func_name = "reciprocal"
 
+    @classmethod
+    def _is_sparse(cls, x):
+        return False
+
 
 @infer_dtype(np.reciprocal)
 def reciprocal(x, out=None, where=None, **kwargs):

maxframe/tensor/arithmetic/rint.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -22,7 +20,7 @@ from .core import TensorUnaryOp
 from .utils import arithmetic_operator
 
 
-@arithmetic_operator(sparse_mode="unary")
+@arithmetic_operator(sparse_mode="always_false")
 class TensorRint(TensorUnaryOp):
     _op_type_ = opcodes.RINT
     _func_name = "rint"

maxframe/tensor/arithmetic/rshift.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/setimag.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/setreal.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/sign.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/signbit.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/sin.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/sinc.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -22,7 +20,7 @@ from .core import TensorUnaryOp
 from .utils import arithmetic_operator
 
 
-@arithmetic_operator(sparse_mode="unary")
+@arithmetic_operator(sparse_mode="always_false")
 class TensorSinc(TensorUnaryOp):
     _op_type_ = opcodes.SINC
     _func_name = "sinc"

maxframe/tensor/arithmetic/sinh.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/spacing.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/sqrt.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/square.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/subtract.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -27,6 +25,10 @@ class TensorSubtract(TensorBinOp):
     _op_type_ = opcodes.SUB
     _func_name = "subtract"
 
+    @classmethod
+    def _is_sparse_with_scalar(cls, scalar_val, lhs):
+        return isinstance(scalar_val, (int, float)) and scalar_val == 0
+
 
 @infer_dtype(np.subtract)
 def subtract(x1, x2, out=None, where=None, **kwargs):

maxframe/tensor/arithmetic/tan.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/tanh.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/tests/__init__.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/tests/test_arithmetic.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,14 +14,14 @@
 
 import numpy as np
 import pytest
-
-from maxframe.tensor.arithmetic.core import TensorBinOp, TensorUnaryOp
-from maxframe.utils import collect_leaf_operators
+import scipy.sparse as sps
 
 from ....core import enter_mode
+from ....utils import collect_leaf_operators
 from ...core import SparseTensor, Tensor
 from ...datasource import array, empty, ones, tensor
 from .. import *  # noqa: F401
+from ..core import TensorBinOp, TensorUnaryOp
 
 
 def test_add():
@@ -51,8 +49,8 @@ def test_add():
 
     t = t1 + 1
     assert t.op.gpu is None
-    assert t.issparse() is True
-    assert type(t) is SparseTensor
+    assert t.issparse() is False
+    assert type(t) is Tensor
 
     t = t1 + 0
     assert t.issparse() is True
@@ -343,8 +341,8 @@ def test_cos():
     t1 = tensor([[0, 1, 0], [1, 0, 0]], chunk_size=2).tosparse()
 
     t = cos(t1)
-    assert t.issparse() is True
-    assert type(t) is SparseTensor
+    assert t.issparse() is False
+    assert type(t) is Tensor
 
 
 def test_around():
@@ -416,3 +414,39 @@ def test_binary_op_func_name():
     for op_type in results:
         if op_type not in (TensorSetImag, TensorSetReal):
             assert hasattr(op_type, "_func_name")
+
+
+def test_tree_arithmetic():
+    raws = [np.random.rand(10, 10) for _ in range(10)]
+    tensors = [tensor(a, chunk_size=3) for a in raws]
+
+    t = tree_add(*tensors, combine_size=4)
+    assert isinstance(t.op, TensorTreeAdd)
+    assert t.issparse() is False
+    assert len(t.inputs) == 3
+    assert len(t.inputs[0].inputs) == 4
+    assert len(t.inputs[-1].inputs) == 2
+
+    t = tree_multiply(*tensors, combine_size=4)
+    assert isinstance(t.op, TensorTreeMultiply)
+    assert t.issparse() is False
+    assert len(t.inputs) == 3
+    assert len(t.inputs[0].inputs) == 4
+    assert len(t.inputs[-1].inputs) == 2
+
+    raws = [sps.random(5, 9, density=0.1) for _ in range(10)]
+    tensors = [tensor(a, chunk_size=3) for a in raws]
+
+    t = tree_add(*tensors, combine_size=4)
+    assert isinstance(t.op, TensorTreeAdd)
+    assert t.issparse() is True
+    assert len(t.inputs) == 3
+    assert len(t.inputs[0].inputs) == 4
+    assert len(t.inputs[-1].inputs) == 2
+
+    t = tree_multiply(*tensors, combine_size=4)
+    assert isinstance(t.op, TensorTreeMultiply)
+    assert t.issparse() is True
+    assert len(t.inputs) == 3
+    assert len(t.inputs[0].inputs) == 4
+    assert len(t.inputs[-1].inputs) == 2

maxframe/tensor/arithmetic/truediv.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/trunc.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/arithmetic/utils.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,6 +14,8 @@
 
 import numpy as np
 
+from ...config import options
+
 
 def arithmetic_operator(cls=None, init=True, sparse_mode=None):
     def _decorator(cls):
@@ -26,11 +26,12 @@ def arithmetic_operator(cls=None, init=True, sparse_mode=None):
         def _is_sparse_binary_and_const(x1, x2):
             if all(np.isscalar(x) for x in [x1, x2]):
                 return False
-            if all(
-                np.isscalar(x) or (hasattr(x, "issparse") and x.issparse())
-                for x in [x1, x2]
-            ):
+            if all(hasattr(x, "issparse") and x.issparse() for x in [x1, x2]):
                 return True
+            if np.isscalar(x1):
+                return cls._is_sparse_with_scalar(x1, True)
+            if np.isscalar(x2):
+                return cls._is_sparse_with_scalar(x2, False)
             return False
 
         def _is_sparse_binary_or_const(x1, x2):
@@ -63,3 +64,28 @@ def arithmetic_operator(cls=None, init=True, sparse_mode=None):
         return _decorator(cls)
     else:
         return _decorator
+
+
+class TreeReductionBuilder:
+    def __init__(self, combine_size=None):
+        self._combine_size = combine_size or options.dpe.reduction.combine_size
+
+    def _build_reduction(self, inputs, final=False):
+        raise NotImplementedError
+
+    def build(self, inputs):
+        combine_size = self._combine_size
+        while len(inputs) > self._combine_size:
+            new_inputs = []
+            for i in range(0, len(inputs), combine_size):
+                objs = inputs[i : i + combine_size]
+                if len(objs) == 1:
+                    obj = objs[0]
+                else:
+                    obj = self._build_reduction(objs, final=False)
+                new_inputs.append(obj)
+            inputs = new_inputs
+
+        if len(inputs) == 1:
+            return inputs[0]
+        return self._build_reduction(inputs, final=True)

maxframe/tensor/array_utils.py

@@ -107,25 +107,13 @@ def _is_array_writeable(a):
     return True
 
 
-def as_same_device(inputs, device=None, ret_extra=False, copy_if_not_writeable=False):
+def as_same_module(inputs, ret_extra=False, copy_if_not_writeable=False):
     input_tensors = [
         i for i in inputs if hasattr(i, "ndim") and i.ndim > 0
     ]  # filter scalar
     has_sparse = any(issparse(i) for i in inputs)
 
-    if device is None:
-        try:
-            device = _most_nbytes_device(
-                (i.device.id if hasattr(i, "device") else -1, i.nbytes)
-                for i in input_tensors
-            )
-        except ValueError:
-            device = -1
-
-    if device == -1:
-        outputs = [_get(i) for i in inputs]
-    else:
-        outputs = [move_to_device(i, device) for i in inputs]
+    outputs = [_get(i) for i in inputs]
 
     if copy_if_not_writeable:
         new_outputs = []
@@ -155,7 +143,7 @@ def as_same_device(inputs, device=None, ret_extra=False, copy_if_not_writeable=F
             m = get_array_module(input_tensors[0])
         else:
             m = np
-    return outputs, device, m
+    return outputs, m
 
 
 def as_np_array(x):
@@ -174,13 +162,3 @@ def device(device_id):
     else:  # pragma: no cover
         with cp.cuda.Device(device_id) as dev:
             yield dev
-
-
-def create_array(op):
-    xp = array_module(op.gpu)
-
-    def inner(func, *args, **kwargs):
-        with device(op.device):
-            return getattr(xp, func)(*args, **kwargs)
-
-    return inner

maxframe/tensor/core.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -30,7 +28,7 @@ from ..core import (
     is_build_mode,
     register_output_types,
 )
-from ..core.entity.utils import refresh_tileable_shape
+from ..core.entity.utils import fill_chunk_slices, refresh_tileable_shape
 from ..serialization.serializables import (
     AnyField,
     DataTypeField,
@@ -50,7 +48,7 @@ class TensorOrder(Enum):
 
 
 class TensorData(HasShapeTileableData, _ExecuteAndFetchMixin):
-    __slots__ = ()
+    __slots__ = ("_accessors",)
     type_name = "Tensor"
 
     _legacy_deprecated_non_primitives = ["_chunks"]
@@ -79,6 +77,7 @@ class TensorData(HasShapeTileableData, _ExecuteAndFetchMixin):
             _order=order,
             **kw,
         )
+        self._accessors = dict()
         if self.order is None and self.op is not None:
             if len(self.inputs) == 0:
                 self._order = TensorOrder.C_ORDER
@@ -137,7 +136,8 @@ class TensorData(HasShapeTileableData, _ExecuteAndFetchMixin):
 
     def refresh_params(self):
         refresh_tileable_shape(self)
-        if self._dtype is None:
+        fill_chunk_slices(self)
+        if self.dtype is None:
             self._dtype = self.chunks[0].dtype
 
     @property
@@ -458,7 +458,7 @@ class Tensor(HasShapeTileable):
         array([('c', 1), ('a', 2)],
               dtype=[('x', '|S1'), ('y', '<i4')])
         """
-        from .misc import sort
+        from .sort import sort
 
         self._data = sort(
             self,

maxframe/tensor/datasource/__init__.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .arange import arange
+from .arange import TensorArange, arange
 from .array import (
     ArrayDataSource,
     CSRMatrixDataSource,
@@ -22,11 +22,19 @@ from .array import (
     asfortranarray,
     tensor,
 )
+from .diag import TensorDiag, diag
+from .diagflat import diagflat
 from .empty import TensorEmpty, TensorEmptyLike, empty, empty_like
+from .eye import TensorEye, eye
 from .from_dataframe import from_dataframe, from_index, from_series
 from .from_dense import fromdense
 from .from_sparse import fromsparse
 from .full import full, full_like
+from .identity import identity
+from .indices import indices
+from .linspace import TensorLinspace, linspace
+from .meshgrid import meshgrid
 from .ones import TensorOnes, TensorOnesLike, ones, ones_like
 from .scalar import Scalar, scalar
-from .zeros import TensorZeros, zeros
+from .tri_array import TensorTril, TensorTriu, tril, triu
+from .zeros import TensorZeros, TensorZerosLike, zeros, zeros_like

maxframe/tensor/datasource/arange.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 # Copyright 1999-2025 Alibaba Group Holding Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

maxframe/tensor/datasource/array.py

@@ -15,7 +15,7 @@
 import numpy as np
 
 from ... import opcodes
-from ...lib.sparse.core import cp, cps, get_array_module, issparse, sps
+from ...lib.sparse.core import cp, get_array_module, issparse
 from ...serialization.serializables import (
     AnyField,
     FieldTypes,
@@ -26,7 +26,6 @@ from ...serialization.serializables import (
 from ...utils import on_deserialize_shape, on_serialize_shape
 from ..array_utils import is_array, is_cupy
 from ..core import TENSOR_TYPE, Tensor, TensorData, TensorOrder
-from ..utils import get_chunk_slices
 from .core import TensorNoInput
 from .scalar import scalar
 
@@ -81,26 +80,6 @@ class CSRMatrixDataSource(TensorNoInput):
             kw["gpu"] = True
         super().__init__(data=data, **kw)
 
-    def to_chunk_op(self, *args):
-        _, idx, chunk_size = args
-
-        xps = cps if self.gpu else sps
-        if len(self.shape) == 1:
-            shape = (1, self.shape[0])
-        else:
-            shape = self.shape
-        data = xps.csr_matrix((self.data, self.indices, self.indptr), shape)
-        chunk_data = data[get_chunk_slices(chunk_size, idx)]
-
-        chunk_op = self.copy().reset_key()
-        chunk_op.data = chunk_data.data
-        chunk_op.indices = chunk_data.indices
-        chunk_op.indptr = chunk_data.indptr
-        chunk_shape = chunk_data.shape[1:] if len(self.shape) == 1 else chunk_data.shape
-        chunk_op.shape = chunk_shape
-
-        return chunk_op
-
 
 def _from_spmatrix(spmatrix, dtype=None, chunk_size=None, gpu=None):
     if gpu is None:
@@ -152,6 +131,8 @@ def tensor(
         try:
             data = m.asarray(data, dtype=dtype, order=order)
         except ValueError:
+            if not hasattr(data, "__array__"):
+                raise
             arr = data.__array__(dtype=dtype)
             if isinstance(arr, TENSOR_TYPE):
                 return arr.astype(arr.dtype, order=order, copy=False)