template<typename AFloat>
class TMVA::DNN::TCpuTensor< AFloat >
Definition at line 40 of file CpuTensor.h.
|
| TCpuTensor () |
|
| TCpuTensor (AFloat *data, const Shape_t &shape, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
|
| TCpuTensor (const TCpuBuffer< AFloat > &buffer, Shape_t shape, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from a TCpuBuffer and a shape
|
|
| TCpuTensor (const TCpuMatrix< AFloat > &matrix, size_t dim=3, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from a TCpuMatrix.
|
|
| TCpuTensor (Shape_t shape, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from a shape.
|
|
| TCpuTensor (size_t bsize, size_t depth, size_t height, size_t width, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from batch size, depth, height, width
|
|
| TCpuTensor (size_t bsize, size_t depth, size_t hw, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from batch size, depth, height*width
|
|
| TCpuTensor (size_t n, size_t m, MemoryLayout memlayout=MemoryLayout::ColumnMajor) |
| constructors from n m
|
|
TCpuTensor< AFloat > | At (size_t i) |
|
TCpuTensor< AFloat > | At (size_t i) const |
|
size_t | GetBufferUseCount () const |
|
size_t | GetCSize () const |
|
TCpuBuffer< AFloat > & | GetDeviceBuffer () |
|
const TCpuBuffer< AFloat > & | GetDeviceBuffer () const |
|
size_t | GetFirstSize () const |
|
size_t | GetHSize () const |
|
MemoryLayout | GetLayout () const |
|
TCpuMatrix< AFloat > | GetMatrix () const |
|
size_t | GetNcols () const |
|
size_t | GetNoElements () const |
|
size_t | GetNrows () const |
|
AFloat * | GetRawDataPointer () |
| Return raw pointer to the elements stored contiguously in column-major order.
|
|
const AFloat * | GetRawDataPointer () const |
|
size_t | GetWSize () const |
|
template<typename Function_t > |
void | Map (Function_t &f) |
| Map the given function over the matrix elements.
|
|
template<typename Function_t > |
void | MapFrom (Function_t &f, const TCpuTensor< AFloat > &A) |
| Same as maps but takes the input values from the tensor A and writes the results in this tensor.
|
|
| operator TMatrixT () const |
| Convert to a TMatrixT<AFloat_t> object.
|
|
AFloat & | operator() (size_t i, size_t j) |
|
AFloat | operator() (size_t i, size_t j) const |
|
AFloat & | operator() (size_t i, size_t j, size_t k) |
|
AFloat | operator() (size_t i, size_t j, size_t k) const |
|
TCpuMatrix< AFloat > | operator[] (size_t i) const |
|
void | Print (const char *name="Tensor") const |
|
void | PrintShape (const char *name="Tensor") const |
|
TCpuTensor< AFloat > | Reshape (Shape_t shape) const |
|
void | Zero () |
|
| RTensor (Shape_t shape, MemoryLayout layout=MemoryLayout::RowMajor) |
| Construct a tensor owning data initialized with new container.
|
|
| RTensor (std::shared_ptr< Container_t > container, Shape_t shape, MemoryLayout layout=MemoryLayout::RowMajor) |
| Construct a tensor owning externally provided data.
|
|
| RTensor (Value_t *data, Shape_t shape, MemoryLayout layout=MemoryLayout::RowMajor) |
| Construct a tensor as view on data.
|
|
| RTensor (Value_t *data, Shape_t shape, Shape_t strides, MemoryLayout layout=MemoryLayout::RowMajor) |
| Construct a tensor as view on data.
|
|
Iterator | begin () noexcept |
|
RTensor< Value_t, Container_t > | Copy (MemoryLayout layout=MemoryLayout::RowMajor) const |
| Copy RTensor to new object.
|
|
Iterator | end () noexcept |
|
RTensor< Value_t, Container_t > | ExpandDims (int idx) const |
| Expand dimensions.
|
|
std::shared_ptr< Container_t > | GetContainer () |
|
const std::shared_ptr< Container_t > | GetContainer () const |
|
Value_t * | GetData () |
|
const Value_t * | GetData () const |
|
MemoryLayout | GetMemoryLayout () const |
|
const Shape_t & | GetShape () const |
|
std::size_t | GetSize () const |
|
const Shape_t & | GetStrides () const |
|
bool | IsOwner () const |
|
bool | IsView () const |
|
Value_t & | operator() (const Index_t &idx) |
| Access elements.
|
|
const Value_t & | operator() (const Index_t &idx) const |
| Access elements.
|
|
Value_t & | operator() (Idx... idx) |
| Access elements.
|
|
const Value_t & | operator() (Idx... idx) const |
| Access elements.
|
|
RTensor< Value_t, Container_t > | Reshape (const Shape_t &shape) const |
| Reshape tensor.
|
|
RTensor< Value_t, Container_t > | Resize (const Shape_t &shape) |
| Resize tensor.
|
|
RTensor< Value_t, Container_t > | Slice (const Slice_t &slice) |
| Create a slice of the tensor.
|
|
RTensor< Value_t, Container_t > | Squeeze () const |
| Squeeze dimensions.
|
|
RTensor< Value_t, Container_t > | Transpose () const |
| Transpose.
|
|