Create Tensorflow variable with weight regularization

gistfile1.py

def variable_on_cpu(self, name, shape, initializer):
  with tf.device('/cpu:0'):
    var = tf.get_variable(name, shape, initializer=initializer)
  return var

def variable_with_weight_decay(self, name, shape, stddev, wd):
  var = self.variable_on_cpu(name, shape,
                        tf.truncated_normal_initializer(stddev=stddev))
  if wd:
    weight_decay = tf.mul(tf.nn.l2_loss(var), wd, name='weight_loss')
    tf.add_to_collection('losses', weight_decay)
  return var