Update MMD_RBF.py

YangYu-NUAA · Mar 28, 2023 · 36d557e · 36d557e
1 parent 1e643af
commit 36d557e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/MMD_RBF.py b/MMD_RBF.py
@@ -1,3 +1,7 @@
+import numpy as np
+import torch
+import torch.nn.functional as F
+
 def mmd_rbf(source, target, kernel_mul=2.0, kernel_num=5, fix_sigma=None):
     '''
     计算MMD距离，使用RBF核函数