entropy_ops

Ops for computing two-body entropies.

---

source

smooth_mean_one_body_entropy

 smooth_mean_one_body_entropy (x:torch.Tensor)

Smooth extension of the plug-in estimator of the onr-body Shannon entropy. x must have shape (…, N, L, R). The result has shape (…,).

---

source

pointwise_shannon

 pointwise_shannon (ps, eps=1e-20)

---

source

smooth_mean_two_body_entropy

 smooth_mean_two_body_entropy (x:torch.Tensor, y:torch.Tensor)

Smooth extension of the plug-in estimator of the two-body Shannon entropy. x must have shape (…, N, L, R), and y must have shape (N, L, R). The result has shape (…,).