xnn.cnn.models.spherical.so3_near_identity_grid#

xnn.cnn.models.spherical.so3_near_identity_grid(max_beta=0.39269908169872414, max_gamma=6.283185307179586, n_alpha=8, n_beta=3, n_gamma=None)[source]#

Rotations (beta, alpha, gamma) on rings around the identity: every rotation of a ring is at the same distance from the identity.

Returns:

Shape (n_alpha * n_beta * n_gamma, 3), float64.

Return type:

Tensor

Parameters:

max_beta (float)
max_gamma (float)
n_alpha (int)
n_beta (int)
n_gamma (int | None)