• shivadbhavsar's avatar
    scalar unsqueeze broadcast support (#1753) · 2140fe19
    shivadbhavsar authored
    Adding support for broadcasted scalars to unsqueeze op.
    
    Specifying steps other than 1 is disallowed in this implementation since we want the output the always be a tensor. We can support varying step sizes if we allow a broadcasted scalar output from this op.
    2140fe19
op_shape_test.cpp 131 KB