e2403d010270eae28c51f81900e763dfeba1f16d,experiments/custom_tdnn_layer.py,TimeDelayLayer,build,#TimeDelayLayer#,13

Before Change

    def build(self, input_shape):
        // Create a trainable weight variable for this layer.
        self.kernel = self.add_weight(name="kernel",
                                      shape=(input_shape[1], self.output_dim),
                                      initializer="uniform",
                                      trainable=True)
        super(TimeDelayLayer, self).build(input_shape)  // Be sure to call this somewhere!

After Change

        // input_shape traditionally is (batch_size, steps, input_dim)
        input_dim = input_shape[2:]
        print ("In build method...")
        print ("input_shape: " + str(input_shape))
        print ("input_dim: " + str(input_dim))

        // Create a trainable weight variable for this layer.

Instances