def activation_features(reshape, scale_factor=1.0):
    """Returns image data as features."""

    def generate(data):
        if reshape:
            batch_size, num_channels = data.shape[0:2]
            data = data.reshape([batch_size, num_channels, -1])
        return scale_factor * data

    return generate