Call internal::array_prod to compute the total size of the tensor.

This commit is contained in:
Benoit Steiner 2016-11-28 09:00:31 -08:00
parent 9fd081cddc
commit 3011dc94ef

View File

@ -21,7 +21,7 @@ static void test_0d()
TensorFixedSize<float, Sizes<>, RowMajor> scalar2; TensorFixedSize<float, Sizes<>, RowMajor> scalar2;
VERIFY_IS_EQUAL(scalar1.rank(), 0); VERIFY_IS_EQUAL(scalar1.rank(), 0);
VERIFY_IS_EQUAL(scalar1.size(), 1); VERIFY_IS_EQUAL(scalar1.size(), 1);
VERIFY_IS_EQUAL(array_prod(scalar1.dimensions()), 1); VERIFY_IS_EQUAL(internal::array_prod(scalar1.dimensions()), 1);
scalar1() = 7.0; scalar1() = 7.0;
scalar2() = 13.0; scalar2() = 13.0;