def distance(self, A, B):
return torch.abs(torch.mean(A) - torch.mean(B))
微信
公众号
扫码关注公众号
微信
公众号