create_family_torch: Function to create (custom) family
Description
Function to create (custom) family
Usage
create_family_torch(torch_dist, trafo_list, output_dim = 1L)
Value
a function that can be used to train a
distribution learning model in torch
Arguments
- torch_dist
a torch probability distribution
- trafo_list
list of transformations h for each parameter
(e.g, exp
for a variance parameter)
- output_dim
integer defining the size of the response