tensor_blob.h File Reference

TBlob class that holds common representation of arbirary dimension tensor, can be used to transformed to normal fixed dimenson tensor. More...

#include <dmlc/logging.h>
#include <dmlc/json.h>
#include <dlpack/dlpack.h>
#include <vector>
#include <iostream>
#include <utility>
#include <algorithm>
#include "./base.h"

Include dependency graph for tensor_blob.h:

Go to the source code of this file.

Classes
class	mxnet::TBlob
	tensor blob class that can be used to hold tensor of any dimension, any device and any data type, This is a weak type that can be used to transfer data through interface TBlob itself do not involve any arithmentic operations, but it can be converted to tensor of fixed dimension for further operations More...
class	dmlc::parameter::FieldEntry< mxnet::TShape >

Namespaces
	mxnet
	namespace of mxnet
	dmlc
	dmlc::parameter

Functions
	dmlc::DMLC_DECLARE_TYPE_NAME (mxnet::TShape,"Shape(tuple)")
	dmlc::DMLC_DECLARE_TYPE_NAME (mxnet::Tuple< int >,"Shape(tuple)")
	dmlc::DMLC_DECLARE_TYPE_NAME (mxnet::Tuple< dmlc::optional< int >>,"Shape(tuple)")
	dmlc::DMLC_DECLARE_TYPE_NAME (nnvm::Tuple< int >,"Shape(tuple)")
	dmlc::DMLC_DECLARE_TYPE_NAME (nnvm::Tuple< dmlc::optional< int >>,"Shape(tuple)")

Variables
constexpr const int	mxnet::kCPU = kDLCPU
constexpr const int	mxnet::kGPU = kDLGPU
constexpr const int	mxnet::kTVMNDArrayTypeCode = 19

Detailed Description

TBlob class that holds common representation of arbirary dimension tensor, can be used to transformed to normal fixed dimenson tensor.

Copyright (c) 2014 by Contributors

Author

Tianqi Chen