Go to the source code of this file.
|
template<typename T > |
void | magmadnn::math::tile (Tensor< T > *A, Tensor< T > *B, unsigned int t, unsigned int axis) |
|
- Author
- Sedrick Keh
- Version
- 0.1
- Date
- 2019-06-17
- Copyright
- Copyright (c) 2019
◆ 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 | |
B | axis dim should equal A axis dim * t |