I am trying to generate a random matrix that is filled with a discrete distribution with three values. In my case, my distribution looks like -1 with probability 1/4, +1 with probability -1/4, and 0 with probability 1/2. Is there an elegant way to do this in Pytorch? I am working with the C++ frontend, but would be willing to switch to the python frontend to get this working.