Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Macros
Public Member Functions | Public Attributes | List of all members
mshadow::expr::SwapAxisExp< SrcExp, dimsrc, 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 <tensor_expr_ext.h>

Inheritance diagram for mshadow::expr::SwapAxisExp< SrcExp, dimsrc, a1, a2 >:
mshadow::expr::MakeTensorExp< SwapAxisExp< SrcExp, dimsrc, a1, a2 >, SrcExp, dimsrc > mshadow::expr::Exp< MakeTensorExp< SwapAxisExp< SrcExp, dimsrc, a1, a2 >, SrcExp, dim >, type::kMapper >

Public Member Functions

 SwapAxisExp (const SrcExp &src)
 constructor
 
- Public Member Functions inherited from mshadow::expr::MakeTensorExp< SwapAxisExp< SrcExp, dimsrc, a1, a2 >, SrcExp, dimsrc >
const SwapAxisExp< SrcExp,
dimsrc, a1, a2 > & 
real_self (void) const
 true self of subtype
 
- Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< SwapAxisExp< SrcExp, dimsrc, a1, a2 >, SrcExp, dim >, type::kMapper >
const MakeTensorExp
< SwapAxisExp< SrcExp, dimsrc,
a1, a2 >, SrcExp, dim > & 
self (void) const
 
MakeTensorExp< SwapAxisExp
< SrcExp, dimsrc, a1, a2 >
, SrcExp, dim > & 
refself (void)
 

Public Attributes

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

Detailed Description

template<typename SrcExp, int dimsrc, int a1, int a2>
struct mshadow::expr::SwapAxisExp< SrcExp, dimsrc, 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
dimsrcsource dimension
a1smaller dimension to be swapped
a2larger dimension to be swapped

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