|
ROOT 6.18/05 Reference Guide |
Class representing training batches consisting of a vector of matrices as input data and a matrix of output data. The input and output data can be accessed using the GetInput() and GetOutput() member functions.
| Architecture_t | The underlying architecture. |
Definition at line 57 of file TensorDataLoader.h.
Public Types | |
| using | Matrix_t = typename Architecture_t::Matrix_t |
Public Member Functions | |
| TTensorBatch (const TTensorBatch &)=default | |
| TTensorBatch (std::vector< Matrix_t > &, Matrix_t &, Matrix_t &) | |
| TTensorBatch (TTensorBatch &&)=default | |
| std::vector< Matrix_t > & | GetInput () |
| Return the tensor representing the input data. More... | |
| Matrix_t & | GetOutput () |
| Return the matrix representing the output data. More... | |
| Matrix_t & | GetWeights () |
| Return the matrix holding the event weights. More... | |
| TTensorBatch & | operator= (const TTensorBatch &)=default |
| TTensorBatch & | operator= (TTensorBatch &&)=default |
Private Attributes | |
| std::vector< Matrix_t > | fInputTensor |
| The input tensor batch, one matrix one input. More... | |
| Matrix_t | fOutputMatrix |
| The output matrix representing the ground truth. More... | |
| Matrix_t | fWeightMatrix |
#include <TMVA/DNN/TensorDataLoader.h>
| using TMVA::DNN::TTensorBatch< Architecture_t >::Matrix_t = typename Architecture_t::Matrix_t |
Definition at line 59 of file TensorDataLoader.h.
| TMVA::DNN::TTensorBatch< Architecture_t >::TTensorBatch | ( | std::vector< Matrix_t > & | inputTensor, |
| Matrix_t & | outputMatrix, | ||
| Matrix_t & | weightMatrix | ||
| ) |
Definition at line 192 of file TensorDataLoader.h.
|
default |
|
default |
|
inline |
Return the tensor representing the input data.
Definition at line 74 of file TensorDataLoader.h.
|
inline |
Return the matrix representing the output data.
Definition at line 76 of file TensorDataLoader.h.
|
inline |
Return the matrix holding the event weights.
Definition at line 78 of file TensorDataLoader.h.
|
default |
|
default |
|
private |
The input tensor batch, one matrix one input.
Definition at line 62 of file TensorDataLoader.h.
|
private |
The output matrix representing the ground truth.
Definition at line 63 of file TensorDataLoader.h.
|
private |
Definition at line 64 of file TensorDataLoader.h.