如何为dymos ODE定义向量输入？

在所有我见过的Dymos示例中，输入的定义形式如下：

self.add_input('name', val=np.ones(num_nodes), ...)

这里的num_nodes指的是转录方案中的节点数量。

我想请教：是否可以改用如下方式来定义输入？

self.add_input('name', val=np.ones((num_nodes, 3)), ...)

我的使用场景是位移变量包含x、y、z三个分量，不想重复三次调用self.add_input(...)来分别定义每个分量。

内容的提问来源于stack exchange，提问作者aaronfu