https://github.com/sql-machine-learning/elasticdl/blob/5fac16d29854001f8f68d4a0d514f4f57b95b8eb/elasticdl/proto/elasticdl.proto#L118 The proto definition is `TensorFlow` code, and the definition needs to support multiple types. Maybe general style such as `NumPy` is a better choice,