mxnet
|
support for swapaxis More...
Go to the source code of this file.
Classes | |
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 > |
Namespaces | |
mshadow | |
namespace for mshadow | |
mshadow::expr | |
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 | |
Functions | |
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... | |
support for swapaxis
Copyright (c) 2014 by Contributors