Np.unwrap function in pytorch

NOTE: the call to ‘diff’ should have same_size=True