R/mdn.R
mdn_module.Rd
Build the MDN torch module
mdn_module(dim_x, dim_theta, n_components, hidden, embedding = NULL)