Should we use "print_type=True" here? It would give ndim information.