faster_wavenet.py 文件源码

def _forward_softmax_block(self, x_batch, apply_softmax=True):
        input_batch = Variable(x_batch)
        for layer in self.softmax_conv_layers:
            input_batch = F.elu(input_batch)
            output = layer(input_batch)
            input_batch = output
        if apply_softmax:
            output = F.softmax(output)
        return output