MeanMaximumDiscrepancy计算代码,分为线性核和高斯核两种,以pytorch书写的