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