有关于pytorch 计算mutual information

torch并没有提供MI的计算接口,笔者大规模搜罗了网上现有的资料,只找到这些:

  • NPEET: Numpy 向量之间MI计算的package
  • MINE: 利用神将网络估计两个向量之间的MI
  • github 计算图像之间的MI

总结就是没有那种直接能够计算torch两个dense tensor之间MI的开源实现,看来只能用两个向量之间的点积来代替互信息,具体参考:
How to calculate mutual information in PyTorch

你可能感兴趣的:(深度学习,pytorch,深度学习,python)