pub fn injector_net<'a, T : Borrow<Path<'a>>>(params : &Params, vs : T) -> BiConcatThenSequential {
    let vs = vs.borrow();
    let mut net = bi_concat_then_seq();
    let two_matrix_dim = 2 * params.get_flattened_matrix_dim();
    net = net.add(nn::linear(
                     vs / "init_linear",
                     two_matrix_dim as i64,
                     params.num_feat_maps as i64,
                     Default::default()
                  ));
    for i in 0..params.num_injection_layers {
        net = net.add(linear_residual(
                      vs / format!("layer_{}", i),
                      params.num_feat_maps));
    } 
    net
}
///BiModule that takes a flattened matrix (dimension FLATTENED_MATRIX_DIM)
///for an input and a flattened matrix (dimension FLATTENED_MATRIX_DIM) for a target
///to descriptors of NUM_FEAT_MAPS size.
BiModule that takes a flattened matrix (dimension FLATTENED_MATRIX_DIM)
for an input and a flattened matrix (dimension FLATTENED_MATRIX_DIM) for a target
to descriptors of NUM_FEAT_MAPS size.
BiModule that takes a flattened matrix ( dimension FLATTENED_MATRIX_DIM ) for an input and a flattened matrix ( dimension FLATTENED_MATRIX_DIM ) for a target to descriptors of NUM_FEAT_MAPS size .
pub fn injector_net<'a, T : Borrow<Path<'a>>>(params : &Params, vs : T) -> BiConcatThenSequential {
    let vs = vs.borrow();
    let mut net = bi_concat_then_seq();
    let two_matrix_dim = 2 * params.get_flattened_matrix_dim();
    net = net.add(nn::linear(
                     vs / "init_linear",
                     two_matrix_dim as i64,
                     params.num_feat_maps as i64,
                     Default::default()
                  ));
    for i in 0..params.num_injection_layers {
        net = net.add(linear_residual(
                      vs / format!("layer_{}", i),
                      params.num_feat_maps));
    } 
    net
}
pub fn injector_net < ' a , T : Borrow < Path < ' a > > > ( params : & Params , vs : T ) -> BiConcatThenSequential { let vs = vs . borrow ( ) ; let mut net = bi_concat_then_seq ( ) ; let two_matrix_dim = 2 * params . get_flattened_matrix_dim ( ) ; net = net . add ( nn :: linear ( vs / " " , two_matrix_dim as i64 , params . num_feat_maps as i64 , Default :: default ( ) ) ) ; for i in 0 .. params . num_injection_layers { net = net . add ( linear_residual ( vs / format ! ( " " i ) , params . num_feat_maps ) ) ; } net }
BiModule that takes a flattened matrix (dimension FLATTENED_MATRIX_DIM)
for an input and a flattened matrix (dimension FLATTENED_MATRIX_DIM) for a target
to descriptors of NUM_FEAT_MAPS size.
BiModule that takes a flattened matrix ( dimension FLATTENED_MATRIX_DIM ) for an input and a flattened matrix ( dimension FLATTENED_MATRIX_DIM ) for a target to descriptors of NUM_FEAT_MAPS size .
params
&Params
vs
T
params
&Params
vs
T