nitpick: can you add a space between shape and [n_outputs]?