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 |