Function fc_layer

intermediate_source/visualizing_gradients_tutorial.py:61–63

Return a stack of linear->norm->sigmoid layers

(in_size, out_size, norm_layer)

#

def fc_layer(in_size, out_size, norm_layer):
    """Return a stack of linear->norm->sigmoid layers"""
    return nn.Sequential(nn.Linear(in_size, out_size), norm_layer(out_size), nn.Sigmoid())

class Net(nn.Module):
    """Define a network that has num_layers of linear->norm->sigmoid transformations"""