{

if (array.ndim() != 2) {

throw py::value_error(

"Expected 2-dimensional array, got %d"_s.format(array.ndim()));

}

if (array.size() == 0) {

// Sometimes things come through as atleast_2d, etc., but they're empty, so

// don't bother enforcing the trailing shape.

return;

}

if (array.shape(1) != d1) {

throw py::value_error(

"%s must have shape (N, %d), got (%d, %d)"_s.format(

name, d1, array.shape(0), array.shape(1)));

}

{

if (array.ndim() != 3) {

throw py::value_error(

"Expected 3-dimensional array, got %d"_s.format(array.ndim()));

}

if (array.size() == 0) {

// Sometimes things come through as atleast_3d, etc., but they're empty, so

// don't bother enforcing the trailing shape.

return;

}

if (array.shape(1) != d1 || array.shape(2) != d2) {

throw py::value_error(

"%s must have shape (N, %d, %d), got (%d, %d, %d)"_s.format(

name, d1, d2, array.shape(0), array.shape(1), array.shape(2)));

}

{

bool empty = (ND == 0);

for (py::ssize_t i = 0; i < ND; i++) {

if (a.shape(i) == 0) {

empty = true;

}

}

if (empty) {

return 0;

} else {

return a.shape(0);

}