예제 #1
0
def test_wedge_vee_batch():
    phis = torch.Tensor([[1, 2, 3], [4, 5, 6]])
    Phis = SO3.wedge(phis)
    assert (phis == SO3.vee(Phis)).all()
예제 #2
0
def test_wedge_vee():
    phi = torch.Tensor([1, 2, 3])
    Phi = SO3.wedge(phi)
    assert (phi == SO3.vee(Phi)).all()