From f0ddab23de8240c6f6f70cfb9766eaaf36cea471 Mon Sep 17 00:00:00 2001 From: tornadomeet Date: Wed, 28 Oct 2015 09:49:55 +0800 Subject: [PATCH] Update README.md fix small mistakes --- doc/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/doc/README.md b/doc/README.md index 9ea6172f37a7..6fe74210f857 100644 --- a/doc/README.md +++ b/doc/README.md @@ -197,13 +197,13 @@ void ExampleSumRows(void) { - psize_y height of each patch - psize_x width of each patch - pstride: stride of each patch -* return expression type: ```Tensor```, with ```shape=(in_channel*psize*psize, out_height*out_width)```, is Chainer expression +* return expression type: ```Tensor```, with ```shape=(in_channel*psize_x*psize_y, out_height*out_width)```, is Chainer expression ```c++ void ExampleCovolution(Tensor dst, Tensor src, Tensor weight, int ksize, int stride) { int o_height = (src.size(1)- ksize) / stride + 1; int o_width = (src.size(2)- ksize) / stride + 1; - utils::Assert(weight.size(0) == src.size(0) * ksize * ksize); + utils::Assert(weight.size(1) == src.size(0) * ksize * ksize); TensorContainer tmp_col(Shape2(src.size(0) * ksize * ksize, o_height * o_width)); TensorContainer tmp_dst(Shape2(weight.size(0), @@ -230,7 +230,7 @@ void ExampleDecovolution(Tensor bottom, Tensor top, Tensor weight, int ksize, int stride) { int o_height = (bottom.size(1)- ksize) / stride + 1; int o_width = (bottom.size(2)- ksize) / stride + 1; - utils::Assert(weight.size(0) == bottom.size(0) * ksize * ksize); + utils::Assert(weight.size(1) == bottom.size(0) * ksize * ksize); TensorContainer tmp_col(Shape2(bottom.size(0) * ksize * ksize, o_height * o_width)); TensorContainer tmp_dst(Shape2(weight.size(0), o_height*o_width));