concat.h File Reference

#include "tensor/tensor.h"

Include dependency graph for concat.h:

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

Go to the source code of this file.

Functions
template<typename T >
void	magmadnn::math::concat (Tensor< T > *A, Tensor< T > *B, Tensor< T > *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