MagmaDNN  1.0
c++NeuralNetworkFramework
tile.h File Reference
#include "tensor/tensor.h"
Include dependency graph for tile.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::tile (Tensor< T > *A, Tensor< T > *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