07d7e13adcae5fd8dcd8e6a2607f78ba3c1f1dd0,se3cnn/basis_kernels.py,,check_basis_equivariance,#Any#Any#Any#Any#Any#Any#,240

Before Change

    from se3cnn import SO3
    from scipy.ndimage import affine_transform

    n = basis.shape[0]
    dim_in = 2 * order_in + 1
    dim_out = 2 * order_out + 1
    size = basis.shape[-1]
    assert basis.shape == (n, dim_out, dim_in, size, size, size), basis.shape

After Change

    from scipy.ndimage import affine_transform
    import numpy as np

    n = basis.size(0)
    dim_in = 2 * order_in + 1
    dim_out = 2 * order_out + 1
    size = basis.size(-1)
    assert basis.size() == (n, dim_out, dim_in, size, size, size), basis.size()

Instances