增加了dim维数的输入

Signed-off-by: shines77 <gz_shines@msn.com>