15f1d4d4f16a174c0290d6a602418004651a16f4,tensorly/backend/mxnet_backend.py,,,#,10

Before Change

from . import numpy_backend

dtypes = ["int64", "int32", "float32", "float64"]
for dtype in dtypes:
    vars()[dtype] = getattr(numpy, dtype)

def context(tensor):
    Returns the context of a tensor
    
    return {"ctx":tensor.context, "dtype":tensor.dtype}

After Change

from .generic import kron, kr, partial_svd


backend = _generics.new_backend("mxnet")

for name in ["float64", "float32", "int64", "int32"]:
    backend.register(getattr(numpy, name), name=name)

for name in ["arange", "zeros", "zeros_like", "ones", "eye",
             "moveaxis", "dot", "transpose", "reshape",
             "where", "sign", "prod"]:
    backend.register(getattr(nd, name), name=name)

backend.register(kron)
backend.register(kr)
backend.register(partial_svd)


@backend.register

Instances