mxnet
|
#include "../extension.h"
Go to the source code of this file.
Classes | |
struct | mshadow::expr::TakeExp< IndexExp, SrcExp, DType > |
Take a column from a matrix. More... | |
struct | mshadow::expr::Plan< TakeExp< IndexExp, SrcExp, DType >, DType > |
struct | mshadow::expr::ShapeCheck< dim, TakeExp< IndexExp, SrcExp, DType > > |
struct | mshadow::expr::ExpInfo< TakeExp< IndexExp, SrcExp, DType > > |
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 IndexExp , typename SrcExp , typename DType , int e1, int e2> | |
TakeExp< IndexExp, SrcExp, DType > | mshadow::expr::take (const Exp< IndexExp, DType, e1 > &index, const Exp< SrcExp, DType, e2 > &src) |
template<typename IndexExp , typename SrcExp , typename DType > | |
Plan< TakeExp< IndexExp, SrcExp, DType >, DType > | mshadow::expr::MakePlan (const TakeExp< IndexExp, SrcExp, DType > &exp) |
Copyright (c) 2015 by Contributors