将张量分为训练集和测试集

将张量分为训练集和测试集