11.1.2　使用JAX中实现的部件—不要重复造轮子