如何从张量流中的张量构造一个新的分布对象

Question

我想计算CNN的正态分布和张量输出之间的KL偏差。

KL function

论证必须是分布的实例。分配。 所以，我不知道如何从我的张量中构建一个分布实例，这是CNN的输出。