954766f611adb31bb123188691c76c3c7c2c3ea6,paysage/backends/pytorch_backend/matrix.py,,dot,#Any#Any#,227

Before Change

        return b[0, index[0,0]]

def dot(a,b):
    dims = ndim(a) * ndim(b)
    if dims == 4:
        // matrix-matrix product
        return torch.mm(a, b)
    elif dims == 2:
        // matrix-vector product
        return torch.mv(a, b)
    elif dims == 1:
        // vector-vector product
        return torch.dot(a, b)
    else:
        raise ValueError("Cannot determine appropriate matrix product")

def outer(x,y):
    return torch.ger(x, y)

def affine(a,b,W):

After Change

        return b[0, index[0,0]]

def dot(a, b):
    return a @ b

def outer(x,y):
    return torch.ger(x, y)

Instances