ROOT 6.16/01 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.