mxnet
|
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...
#include <transpose.h>
Public Member Functions | |
TransposeIndicesExp (const SrcExp &src_indices, Shape< dimsrc > src_shape, Shape< dimsrc > axes) | |
constructor More... | |
Public Member Functions inherited from mshadow::expr::Exp< TransposeIndicesExp< SrcExp, DType, dimsrc, etype >, DType, etype > | |
const TransposeIndicesExp< SrcExp, DType, dimsrc, etype > & | self (void) const |
TransposeIndicesExp< SrcExp, DType, dimsrc, etype > * | ptrself (void) |
Public Attributes | |
const SrcExp & | src_indices_ |
source expression More... | |
Shape< dimsrc > | src_shape_ |
const Shape< dimsrc > | axes_ |
Shape< dimsrc > | src_in_dst_stride_ |
transform contiguous indices of the source tensor to indices of the transposed tensor. input: Tensor<Device, k>: ishape output: Tensor<Device, k>: oshape = ishape
SrcExp | type of source expression |
DType | the type of elements |
dimsrc | source dimension |
etype | source type |
|
inlineexplicit |
constructor
const Shape<dimsrc> mshadow::expr::TransposeIndicesExp< SrcExp, DType, dimsrc, etype >::axes_ |
Shape<dimsrc> mshadow::expr::TransposeIndicesExp< SrcExp, DType, dimsrc, etype >::src_in_dst_stride_ |
const SrcExp& mshadow::expr::TransposeIndicesExp< SrcExp, DType, dimsrc, etype >::src_indices_ |
source expression
Shape<dimsrc> mshadow::expr::TransposeIndicesExp< SrcExp, DType, dimsrc, etype >::src_shape_ |