fixed dockerfile

Author: BigNoobWasTaken

Dockerfile

@@ -11,6 +11,7 @@ RUN apt-get update && apt-get -y install cmake
 
 WORKDIR /binary-tensor
 COPY src/ ./src/
+COPY test/ ./test/
 COPY CMakeLists.txt ./
 COPY Config.cmake.in ./
 WORKDIR /binary-tensor

test/main.cpp

@@ -24,7 +24,7 @@ int main(int argc, char const *argv[])
     Tensor a01 = Tensor(a1);
     Tensor a02 = Tensor(a2);
     auto b = a01 + a02;
-    b.calc_grad();
+    b.calc_grad(ones(b.get_buffer().shape()));
     std::cout << b << std::endl <<
         a01.get_grad() << std::endl <<
         sizeof a1 << std::endl;