You can use weight (output) in both cases. But with formula, you also have a good idea of what the input is. Later on you can pump breast milk to get a better idea of input, but that doesn't work well at the beginning.

If you have a colic baby, all bets are off, of course.