TensorFlowInterface.py 文件源码

def setupOutput(self):
        inputShape = self.input.get_shape()
        convResult = conv2d(self.input,self.W) + self.b

        convResult = tf.reshape(convResult,[-1,self.units]) # flatten reduced image into a vector
        softMaxed = tf.nn.softmax(convResult)
        self.output = tf.reshape(softMaxed,[-1] + inputShape[1:3].as_list() + [self.units])