Flatten

class dragon.vm.caffe.core.layers.Flatten(layer_param)[source]

Flatten the input along the given axes.

Examples:

layer {
  type: "Flatten"
  bottom: "conv5"
  top: "conv5/flatten"
  flatten_param {
    axis: 1
    end_axis: -1
  }
}