toshas/torch_truncnorm

how to define expand method?

Opened this issue · 0 comments

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.