#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"

Include dependency graph for flattenlayer.h:

This graph shows which files directly or indirectly include this file:

Classes
class	magmadnn::layer::FlattenLayer< T >

Functions
template<typename T >
FlattenLayer< T > *	magmadnn::layer::flatten (op::Operation< T > *input)

Detailed Description

Function Documentation

template<typename T >

FlattenLayer< T > * magmadnn::layer::flatten ( op::Operation< T > * input )

A new Flatten layer. Typically used after a convolutional layer. Converts the input tensor into a 2d Tensor by flattening it.

Template Parameters

T numeric

Parameters

input Input to flatten