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
..
2020-10-18 02:38:59 -07:00
2020-10-22 02:23:05 -07:00
2020-10-20 17:16:17 -07:00
2020-08-27 17:15:51 -07:00