Classes | Namespaces | Functions
swapaxis.h File Reference

support for swapaxis More...

#include <algorithm>
#include <utility>
#include "../extension.h"
Include dependency graph for swapaxis.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.


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] More...
struct  mshadow::expr::Plan< SwapAxisExp< SrcExp, DType, dimsrc, m_a1, a2 >, DType >
struct  mshadow::expr::Plan< SwapAxisExp< SrcExp, DType, dimsrc, 1, a2 >, DType >


 overloaded + operator between half_t and bf16_t
 namespace for abstract expressions and expressions template, have no dependency on tensor.h, These data structure takes no charge in computations, they are only used to define operations and represent expression in a symbolic way


template<int a1, int a2, typename SrcExp , typename DType , int etype>
SwapAxisExp< SrcExp, DType, ExpInfo< SrcExp >::kDim, ExpInfo< SrcExp >::kDim-a1, a2 > mshadow::expr::swapaxis (const Exp< SrcExp, DType, etype > &src)
 a expression that reshapes a tensor to another shape More...

Detailed Description

support for swapaxis

Copyright (c) 2014 by Contributors

Tianqi Chen