def update(self, data):
        """Update the statistics with a batch of data.

        # Arguments
            data: tf.Tensor. One batch of data from tf.data.Dataset.
        """
        if self.dtype is None:
            self.dtype = data.dtype
        if self.shape is None:
            self.shape = data.shape.as_list()
        if self.batch_size is None:
            self.batch_size = data.shape.as_list()[0]
        self.num_samples += data.shape.as_list()[0]
Update the statistics with a batch of data.

        # Arguments
            data: tf.Tensor. One batch of data from tf.data.Dataset.
        
autokeras/engine/analyser.py