ntn.py 文件源码

def __init__(self,inp_size,out_size,activation='tanh', **kwargs):
        super(ntn_layer, self).__init__(**kwargs)
        self.k = out_size
        self.d = inp_size
        self.activation = activations.get(activation)
        self.test_out = 0