Naomi Osaka returned to action at the Madrid Open, a month after a disappointing early exit in Miami had her questioning the future of her career.