commit | 4c3b02f31447037b009e80a0251c0ebd0ee95b7c | [log] [tgz] |
---|---|---|
author | Junjie Bai <bai@in.tum.de> | Fri Oct 13 12:10:34 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Fri Oct 13 12:28:22 2017 -0700 |
tree | 5fbf0d5ac9df238d6a9d304c9be86ba5ba454587 | |
parent | c3a9423c7f339bbfd16260f7615a94de51615bd0 [diff] |
Enable Flatten operator to take an arbitrary axis arguemnt Summary: input dimensions up to "axis" will be flattened to the outer dim of output and the remaining input dims will be the inner dim Closes https://github.com/caffe2/caffe2/pull/1330 Reviewed By: dzhulgakov Differential Revision: D6039560 Pulled By: bddppq fbshipit-source-id: e92c30b49a9288feeefc4a639522406e97e149e1
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the Apache 2.0 license. See the NOTICE file for details.