Joint distribution p(x, y)'s marginal probability p(x)
pxy[i, j] indicates p(x=x_i, y=y_i). p(x) is a torch with two values: p(x=x_0) and p(x=x1).Save to the Tensor, px.