File size: 1,378 Bytes
26ce2a9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
=================
liegroups.torch
=================
The PyTorch implementation uses torch.Tensor as the backend linear algebra library, which allows the user to on the GPU or CPU and integrate with other aspects of PyTorch.
This version provides sensible options for batching the transformations themselves, as well as anything they might operate on, and is generally agnostic to the specific Tensor type (e.g., given a torch.cuda.FloatTensor as input, the output will also be a torch.cuda.FloatTensor).
.. autoclass:: liegroups.torch.SO2
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.so2.SO2Matrix
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.SE2
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.se2.SE2Matrix
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.SO3
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.so3.SO3Matrix
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.SE3
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
.. autoclass:: liegroups.torch.se3.SE3Matrix
:members: cpu, cuda, from_numpy, is_cuda, is_pinned, pin_memory
|