提交 1ed31b4e 编写于 作者: H hedaoyuan

Bug fix.

上级 c6e010d0
...@@ -104,11 +104,11 @@ public: ...@@ -104,11 +104,11 @@ public:
protected: protected:
size_t getFilterHeight(const TensorShape& filter) const { size_t getFilterHeight(const TensorShape& filter) const {
filter[filter.ndims() - 2]; return filter[filter.ndims() - 2];
} }
size_t getFilterWidth(const TensorShape& filter) const { size_t getFilterWidth(const TensorShape& filter) const {
filter[filter.ndims() - 1]; return filter[filter.ndims() - 1];
} }
std::vector<size_t> strides_; std::vector<size_t> strides_;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册