예제 #1
0
    def set_layer(self, src, src_dim, dst_dim, actv_func):
        self.weights.append(self.init_weight([src_dim, dst_dim]))
        self.biases.append(self.init_bias([dst_dim]))

        #call Layer class and set acticate function
        layer = Layer(actv_func).switch_layer()
        return layer.set(src, self.weights[-1], self.biases[-1])