2b79f10398105048a22cb9f03062240e50dff84a,tensorly/backend/pytorch_backend.py,,,#,10

Before Change

from torch import qr
from torch import is_tensor

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

// Order 0 tensor, mxnet....
from math import sqrt as scalar_sqrt

// Equivalent functions in pytorch 
maximum = max

After Change

from . import _generics, numpy_backend


backend = _generics.new_backend("pytorch")

for name in ["float64", "float32", "int64", "int32", "is_tensor", "ones",
             "zeros", "zeros_like", "reshape", "eye", "max", "min", "prod",
             "abs", "sqrt", "sign", "where", "qr"]:
    backend.register(getattr(torch, name), name=name)

backend.register(torch.matmul, name="dot")


@backend.register
def context(tensor):

Instances