Merge branch 'shape_axis_check' of https://github.com/jiafatom/pytorch into shape_axis_check