Ejemplos de SO3.vee en Python

Ejemplo n.º 1

def test_wedge_vee_batch():
    phis = torch.Tensor([[1, 2, 3], [4, 5, 6]])
    Phis = SO3.wedge(phis)
    assert (phis == SO3.vee(Phis)).all()

Ejemplo n.º 2

def test_wedge_vee():
    phi = torch.Tensor([1, 2, 3])
    Phi = SO3.wedge(phi)
    assert (phi == SO3.vee(Phi)).all()