• vfdev's avatar
    Unified inputs for `F.rotate` (#2495) · 76662528
    vfdev authored
    * Added code for F_t.rotate with test
    - updated F.affine tests
    
    * Rotate test tolerance to 2%
    
    * Fixes failing test
    
    * Optimized _expanded_affine_grid with a single matmul op
    
    * Recoded _compute_output_size
    76662528
test_transforms.py 70.7 KB