9.3.2　自定义JAX中的向量Tensor