concat.cpp File Reference

#include "math/concat.h"

Include dependency graph for concat.cpp:

Functions
template<typename T >
void	magmadnn::math::concat (Tensor< T > *A, Tensor< T > *B, Tensor< T > *C, unsigned int axis)
template void	magmadnn::math::concat (Tensor< int > *A, Tensor< int > *B, Tensor< int > *C, unsigned int axis)
template void	magmadnn::math::concat (Tensor< float > *A, Tensor< float > *B, Tensor< float > *C, unsigned int axis)
template void	magmadnn::math::concat (Tensor< double > *A, Tensor< double > *B, Tensor< double > *C, unsigned int axis)

Detailed Description

Author

Sedrick Keh

Version

0.1

Date

2019-06-14

Copyright

Copyright (c) 2019

Function Documentation

concat()

template<typename T >

void magmadnn::math::concat	(	Tensor< T > *	A,
		Tensor< T > *	B,
		Tensor< T > *	C,
		unsigned int	axis
	)

Concatenates A and B along axis and puts output into C A, B, C should have at most 1 different dimension

Parameters

A
B
C	axis dim size should equal sum of A and B axis dim size