Crate caffe2op_flatten
source ·Structs
- | Flattens the input tensor into a 2D matrix. If input tensor has shape | $(d_0, d_1, …, d_n)$ then the output will have shape | $\bigl((d_0 * d_1 * … * d_{(axis-1)}), (d_{axis} * d_{(axis+1)} * … * d_n)\bigr)$. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/flatten_op.cc