mxnet
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 > Struct Template Reference

swap two axis of a tensor input: Tensor<Device,dim>: ishape output: Tensor<Device,dimdst> oshape[a1],oshape[a2] = ishape[a2],oshape[a1] More...

#include <swapaxis.h>

Inheritance diagram for mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >:
Inheritance graph
Collaboration diagram for mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >:
Collaboration graph

Public Member Functions

 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)
 

Public Attributes

const SrcExp & src_
 source expression More...
 
- Public Attributes inherited from mshadow::expr::MakeTensorExp< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, SrcExp, dimsrc, DType >
Shape< dim > shape_
 the shape of this expression More...
 

Static Public Attributes

static const int a1 = dimsrc - m_a1
 

Detailed Description

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
SrcExptype of source expression
DTypethe type of elements
dimsrcsource dimension, assert a1 > a2
m_a1one dimension to be swapped, encoded by dimsrc - a1
a2second dimension to be swapped, encoded by a2

Constructor & Destructor Documentation

template<typename SrcExp, typename DType, int dimsrc, int m_a1, int a2>
mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >::SwapAxisExp ( const SrcExp &  src)
inlineexplicit

constructor

Member Data Documentation

template<typename SrcExp, typename DType, int dimsrc, int m_a1, int a2>
const int mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >::a1 = dimsrc - m_a1
static
template<typename SrcExp, typename DType, int dimsrc, int m_a1, int a2>
const SrcExp& mshadow::expr::SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >::src_

source expression


The documentation for this struct was generated from the following file: