Yeah, I think that __init__() would just have to be modified to permit passing the input_dim and active_dims arguments.
Yeah, I think that __init__() would just have to be modified to permit passing the input_dim and active_dims arguments.