support for transpose More...

#include <algorithm>
#include "../extension.h"

Include dependency graph for transpose.h:

This graph shows which files directly or indirectly include this file:

Classes
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] More...

struct	mshadow::expr::Plan< TransposeExExp< SrcExp, DType, dimsrc >, DType >

struct	mshadow::expr::TransposeIndicesExp< SrcExp, DType, dimsrc, etype >
	transform contiguous indices of the source tensor to indices of the transposed tensor. input: Tensor<Device, k>: ishape output: Tensor<Device, k>: oshape = ishape More...

struct	mshadow::expr::Plan< TransposeIndicesExp< SrcExp, DType, dimsrc, etype >, DType >

struct	mshadow::expr::ShapeCheck< dim, TransposeIndicesExp< SrcExp, DType, dimsrc, etype > >

struct	mshadow::expr::ExpInfo< TransposeIndicesExp< SrcExp, DType, dimsrc, etype > >

Namespaces
	mshadow
	overloaded + operator between half_t and bf16_t

	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<typename SrcExp , typename DType , int etype>
TransposeExExp< SrcExp, DType, ExpInfo< SrcExp >::kDim >	mshadow::expr::transpose (const Exp< SrcExp, DType, etype > &src, Shape< ExpInfo< SrcExp >::kDim > axes)
	a expression that reshapes a tensor to another shape More...

template<typename SrcExp , typename DType , int dimsrc, int etype>
TransposeIndicesExp< SrcExp, DType, dimsrc, etype >	mshadow::expr::transpose_indices (const Exp< SrcExp, DType, etype > &src_indices, Shape< dimsrc > src_shape, Shape< dimsrc > axes)
	a expression that reshapes a tensor to another shape More...

template<typename SrcExp , typename DType , int dimsrc, int etype>
Plan< TransposeIndicesExp< SrcExp, DType, dimsrc, etype >, DType >	mshadow::expr::MakePlan (const TransposeIndicesExp< SrcExp, DType, dimsrc, etype > &e)
	make expression More...

Detailed Description

support for transpose

Classes