code: def get_toy_sym(teacher=True, teacher_noise_precision=None):
      
      if teacher:
          net = mx.symbol.Variable('data')
          net = mx.symbol.FullyConnected(data=net, name='teacher_fc1', num_hidden=100)
          net = mx.symbol.Activation(data=net, name='teacher_relu1', act_type="relu")
          net = mx.symbol.FullyConnected(data=net, name='teacher_fc2', num_hidden=1)
          net = mx.symbol.LinearRegressionOutput(data=net, name='teacher_output',
                                                 grad_scale=teacher_noise_precision)
      else:
          net = mx.symbol.Variable('data')
          net = mx.symbol.FullyConnected(data=net, name='student_fc1', num_hidden=100)
          net = mx.symbol.Activation(data=net, name='student_relu1', act_type="relu")
          student_mean = mx.symbol.FullyConnected(data=net, name='student_mean', num_hidden=1)
          student_var = mx.symbol.FullyConnected(data=net, name='student_var', num_hidden=1)
          net = mx.symbol.Group([student_mean, student_var])
      return net
docs: Get toy symbol