589950b1125c4561b3ec7480cc1f71bfb85249a8,opt_einsum/backends/shared.py,,transpose,#Any#Any#,64

Before Change

def transpose(a, axes):
    backend = _CURRENT_BACKEND[0]
    cache = _SHARING_STACK[-1]
    cache["tensor", id(a)] = a

    axes = tuple(axes)

After Change

def transpose(a, axes):
    _save_tensors(a)

    // hash by axes
    axes = tuple(axes)
    key = "transpose", _CURRENT_BACKEND, id(a), axes

Instances