(q, order, epsilon=0, use_gpu=False)
| 138 | |
| 139 | |
| 140 | def qeuler_np(q, order, epsilon=0, use_gpu=False): |
| 141 | if use_gpu: |
| 142 | q = torch.from_numpy(q).cuda().float() |
| 143 | return qeuler(q, order, epsilon).cpu().numpy() |
| 144 | else: |
| 145 | q = torch.from_numpy(q).contiguous().float() |
| 146 | return qeuler(q, order, epsilon).numpy() |
| 147 | |
| 148 | |
| 149 | def qfix(q): |