b41bf85906c38c8e50059f684a0fd3408a0c20fb,se3cnn/util/plot.py,,spherical_surface,#,11

Before Change

    alpha = torch.linspace(0, 2 * math.pi, 2 * n)
    beta_, alpha_ = torch.meshgrid(beta, alpha)
    x = torch.sin(beta_) * torch.cos(alpha_)
    y = torch.sin(beta_) * torch.sin(alpha_)
    z = torch.cos(beta_)

    beta = 0.5 * (beta[1:] + beta[:-1])
    alpha = 0.5 * (alpha[1:] + alpha[:-1])

After Change

    beta = torch.linspace(1e-16, math.pi - 1e-16, 2 * n)
    alpha = torch.linspace(0, 2 * math.pi, 2 * n)
    beta_, alpha_ = torch.meshgrid(beta, alpha)
    x, y, z = angles_to_xyz(beta_, alpha_)

    beta = 0.5 * (beta[1:] + beta[:-1])
    alpha = 0.5 * (alpha[1:] + alpha[:-1])

Instances