small_mlp.py 文件源码

def make_data(num):
    """
    Make data allocates num samples with input dimension
    3 and output dimension of 1.
    """
    inputs = np.random.normal(size=[3, num])
    targets = np.cbrt(np.square(2.5*inputs[0:1, :]) -
                      inputs[1:2, :] * inputs[2:3, :])
    return inputs, targets