Fix contraction test.

This commit is contained in:
Rasmus Munk Larsen 2018-10-08 16:37:07 -07:00
parent e00487f7d2
commit 1a737e1d6a

View File

@ -471,7 +471,8 @@ static void test_tensor_product()
mat1.setRandom();
mat2.setRandom();
Tensor<float, 4, DataLayout> result = mat1.contract(mat2, Eigen::array<DimPair, 0>{{}});
Eigen::array<DimPair, 0> dims;
Tensor<float, 4, DataLayout> result = mat1.contract(mat2, dims);
VERIFY_IS_EQUAL(result.dimension(0), 2);
VERIFY_IS_EQUAL(result.dimension(1), 3);