mxnet
|
Broadcasting the tensor in multiple axes. The dimension of the source tensor in the given axes must be 1. More...
#include <broadcast_with_axis.h>
Public Member Functions | |
template<typename TShape > | |
BroadcastWithMultiAxesExp (const SrcExp &src, const TShape &axes, const TShape &sizes) | |
Public Member Functions inherited from mshadow::expr::MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dimsrc, DType > | |
const BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc > & | real_self (void) const |
true self of subtype More... | |
Public Member Functions inherited from mshadow::expr::Exp< MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType >, DType, type::kChainer > | |
const MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > & | self (void) const |
MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dim, DType > * | ptrself (void) |
Public Attributes | |
const SrcExp & | src_ |
data oprand More... | |
index_t | dst_last_ |
size of the last dimension of dst More... | |
index_t | axesnum_ |
number of broadcasting axes More... | |
Shape< dimsrc > | trailings_ |
product of the dimensions after the broadcasting axses More... | |
Shape< dimsrc > | sizes_ |
new dimension of the broadcasting axes More... | |
index_t | last_ |
size of the last dimension of src More... | |
Public Attributes inherited from mshadow::expr::MakeTensorExp< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, SrcExp, dimsrc, DType > | |
Shape< dim > | shape_ |
the shape of this expression More... | |
Broadcasting the tensor in multiple axes. The dimension of the source tensor in the given axes must be 1.
SrcExp | source expression |
DType | data type |
dimsrc | source dimension |
axesnum | number of broadcasting dimensions |
|
inline |
constructor
index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::axesnum_ |
number of broadcasting axes
index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::dst_last_ |
size of the last dimension of dst
index_t mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::last_ |
size of the last dimension of src
Shape<dimsrc> mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::sizes_ |
new dimension of the broadcasting axes
const SrcExp& mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::src_ |
data oprand
Shape<dimsrc> mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >::trailings_ |
product of the dimensions after the broadcasting axses