Inheritance diagram for magmadnn::layer::FullyConnectedLayer< T >:

Collaboration diagram for magmadnn::layer::FullyConnectedLayer< T >:

Public Member Functions
	FullyConnectedLayer (op::Operation< T > *input, unsigned int hidden_units, bool use_bias=true)

virtual std::vector< op::Operation< T > * >	get_weights ()

op::Operation< T > *	get_weight ()

op::Operation< T > *	get_bias ()

Public Member Functions inherited from magmadnn::layer::Layer< T >
virtual op::Operation< T > *	out ()

op::Operation< T > *	get_input ()

op::Operation< T > *	get_output ()

std::vector< unsigned int >	get_input_shape () const

std::vector< unsigned int >	get_output_shape () const

unsigned int	get_input_shape (unsigned int i) const

unsigned int	get_output_shape (unsigned int i) const

void	set_name (std::string name)

std::string	get_name () const

Protected Member Functions
void	init ()

Protected Member Functions inherited from magmadnn::layer::Layer< T >
	Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input)

Protected Attributes
unsigned int	hidden_units

bool	use_bias

Tensor< T > *	weights_tensor

Tensor< T > *	bias_tensor

op::Operation< T > *	weights

op::Operation< T > *	bias

Protected Attributes inherited from magmadnn::layer::Layer< T >
std::vector< unsigned int >	input_shape

std::vector< unsigned int >	output_shape

op::Operation< T > *	input

op::Operation< T > *	output

std::string	name

The documentation for this class was generated from the following files:

Public Member Functions

Protected Member Functions