toshas / torch_truncnorm

Truncated Normal Distribution in PyTorch
BSD 3-Clause "New" or "Revised" License
79 stars 13 forks source link

how to define expand method? #11

Open XiankangTang opened 4 months ago

XiankangTang commented 4 months ago

Dear Anton,

I'm currently trying to add the expand function to the truncated normal class you provided, as I need to generate multiple truncated normal distributions in parallel. I've made some attempts, but I haven't found a solution to the problem yet. What do you suggest? I am looking forward to your reply.