pytorch

9797b4e5 - try to fix XLA error: torch_xla/csrc/helpers.cpp:85 : Check failed: min_shape_dim <= dim && dim <= max_shape_dim, when running self.slice(dim, start, stop, step)