Function mapper

examples/SpatialTransformer/mnist-addition.py:197–198

(dp)

    # stack the two digits into two channels, and label it with the sum

    def mapper(dp):
        return [np.stack([dp[0], dp[2]], axis=2), dp[1] + dp[3]]
    ds = MapData(ds, dp)
    ds = BatchData(ds, 128)
    return ds