Solved by albanD in post #6 
              
              
                Hi, 
I’m afraid the softplus function only accepts a single beta value. But you can just re-implement it 
batch_size = x.size(0)
# Generate one beta per sample
beta = torch.empty(batch_size).uniform_(self.beta_lower, self.beta_higher)
softplus = ((beta * x).exp() + 1).log() / beta
res = x * torch.ta…