MagmaDNN  1.0
c++NeuralNetworkFramework
tile.cpp File Reference
#include "math/tile.h"
Include dependency graph for tile.cpp:

Functions

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

Detailed Description

Author
Sedrick Keh
Version
0.1
Date
2019-06-17

Function Documentation

◆ tile()

template<typename T >
void magmadnn::math::tile ( Tensor< T > *  A,
Tensor< T > *  B,
unsigned int  t,
unsigned int  axis 
)

Tiles tensor A along axis t times, output into B A, B should have at most 1 different dimension

Parameters
A
Baxis dim should equal A axis dim * t