Jeremy Reizenstein 7e986cfba8 Avoid torch.square ...

Summary: Fix axis_angle conversions where I used torch.square which doesn't work with pytorch 1.4

Reviewed By: nikhilaravi

Differential Revision: D24451546

fbshipit-source-id: ba26f7dad5fa991f0a8f7d3d09ee7151163aecf4

2020-10-22 02:23:05 -07:00

..

__init__.py

Address black + isort fbsource linter warnings

2020-03-29 14:51:02 -07:00

rotation_conversions.py

Avoid torch.square

2020-10-22 02:23:05 -07:00

so3.py

6D representation of rotations.

2020-07-08 04:01:22 -07:00

transform3d.py

Add pyre typeshed information for Tensor.ndim and nn.ConvTranspose2d

2020-09-02 06:52:30 -07:00