swap two axis of a tensor input: Tensor<Device,dim>: ishape output: Tensor<Device,dimdst> oshape[a1],oshape[a2] = ishape[a2],oshape[a1]
More...
|
| SwapAxisExp (const SrcExp &src) |
| constructor More...
|
|
Public Member Functions inherited from mshadow::expr::MakeTensorExp< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, SrcExp, dimsrc, DType > |
const SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 > & | real_self (void) const |
| true self of subtype More...
|
|
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, SrcExp, dim, DType >, DType, type::kChainer > |
const MakeTensorExp< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, SrcExp, dim, DType > & | self (void) const |
|
MakeTensorExp< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, SrcExp, dim, DType > * | ptrself (void) |
|
template<typename SrcExp, typename DType, int dimsrc, int m_a1, int a2>
struct mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >
swap two axis of a tensor input: Tensor<Device,dim>: ishape output: Tensor<Device,dimdst> oshape[a1],oshape[a2] = ishape[a2],oshape[a1]
- Template Parameters
-
SrcExp | type of source expression |
DType | the type of elements |
dimsrc | source dimension, assert a1 > a2 |
m_a1 | one dimension to be swapped, encoded by dimsrc - a1 |
a2 | second dimension to be swapped, encoded by a2 |