Method compute_output_shape

model_demo/DeepCrossing/layer.py:28–29

(self, input_shape)

            return sgx.e_sigmoid(sgx.e_matmul(inputs, self.w))

    def compute_output_shape(self, input_shape):
        return (input_shape[0], self.units)

    def get_config(self):
        config = super(MyDense, self).get_config()