pytorch

3248e0a9 - add relevant test and remove masked_scatter skips