问题：

DTensor 参数初始化时，发现dim_map[shard_dim] index越界。

原因：

dim_map 元祖构建的dim维度取错了，期望是Tensor shape的ndim，实际取成了devicemesh的ndim

解决：

改成shape的dim