import tensorflow as tf

                            self,
                            'kappa_%s' % layer,
                            tf.get_variable(
                                name='%s_kappa' % self.layer_name,
                                dtype=self.dtype,
                                initializer=initialization.xavier_initializer(
                                    shape=bias_shape,
                                    dtype=self.dtype,
                                    uniform=self.normal_initializer,
                                    mask=None)))
                    if self.lesion_omega:
                        setattr(
                            self,
                            'omega_%s' % layer,
                            tf.constant(0.))
                    else:
                        setattr(
                            self,
                            'omega_%s' % layer,
                            tf.get_variable(
                                name='%s_omega' % self.layer_name,
                                dtype=self.dtype,
                                initializer=initialization.xavier_initializer(
                                    shape=bias_shape,
                                    dtype=self.dtype,
                                    uniform=self.normal_initializer,
                                    mask=None)))
                else:
