def METHOD_NAME(generator, ones, rate, training=None, count=1):
    def dropped_inputs():
        return generator.dropout(ones, rate)
    if count > 1:
        return [
            backend.in_train_phase(dropped_inputs, ones, training=training)
            for _ in range(count)
        ]
    return backend.in_train_phase(dropped_inputs, ones, training=training)