transpose axes of a tensor input: Tensor<Device,dim>: ishape output: Tensor<Device,dimdst> oshape[a1],oshape[a2] = ishape[a2],oshape[a1]
More...
|
| TransposeExExp (const SrcExp &src, Shape< dimsrc > axes) |
| constructor More...
|
|
Public Member Functions inherited from mshadow::expr::MakeTensorExp< TransposeExExp< SrcExp, DType, dimsrc >, SrcExp, dimsrc, DType > |
const TransposeExExp< SrcExp, DType, dimsrc > & | real_self (void) const |
| true self of subtype More...
|
|
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< TransposeExExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType >, DType, type::kChainer > |
const MakeTensorExp< TransposeExExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > & | self (void) const |
|
MakeTensorExp< TransposeExExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > * | ptrself (void) |
|
template<typename SrcExp, typename DType, int dimsrc>
struct mshadow::expr::TransposeExExp< SrcExp, DType, dimsrc >
transpose axes 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 |