深度学习数据处理样本均衡问题

深度学习随机下采样中的X为高维数据，不能直接用randomundersampler.fit_resample(X,Y)

代码

from imblearn.under_sampling import RandomUnderSampler # 样本均衡 欠采样
rus = RandomUnderSampler(random_state=0)
temp1, temp2 = rus.fit_resample(Y, Y)
index_rus = rus.sample_indices_#获取均衡后的索引
train_data = X[index_rus]#根据索引取数据
train_label = Y[index_rus]#根据索引取标签

问题

rus.fit_resample(Y, Y),这里传入的是两个二维的数据标签Y,可以这样用吗？