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 59 of file TensorDataLoader.h.
Public Types | |
using | Matrix_t = typename Architecture_t::Matrix_t |
using | Tensor_t = typename Architecture_t::Tensor_t |
Public Member Functions | |
TTensorBatch (const TTensorBatch &)=default | |
TTensorBatch (Tensor_t &, Matrix_t &, Matrix_t &) | |
TTensorBatch (TTensorBatch &&)=default | |
Tensor_t & | GetInput () |
Return the tensor representing the input data. | |
Matrix_t & | GetOutput () |
Return the matrix representing the output data. | |
Matrix_t & | GetWeights () |
Return the matrix holding the event weights. | |
TTensorBatch & | operator= (const TTensorBatch &)=default |
TTensorBatch & | operator= (TTensorBatch &&)=default |
Private Attributes | |
Tensor_t | fInputTensor |
The input tensor batch, one matrix one input. | |
Matrix_t | fOutputMatrix |
The output matrix representing the ground truth. | |
Matrix_t | fWeightMatrix |
The event/example weights. | |
#include <TMVA/DNN/TensorDataLoader.h>
using TMVA::DNN::TTensorBatch< Architecture_t >::Matrix_t = typename Architecture_t::Matrix_t |
Definition at line 61 of file TensorDataLoader.h.
using TMVA::DNN::TTensorBatch< Architecture_t >::Tensor_t = typename Architecture_t::Tensor_t |
Definition at line 62 of file TensorDataLoader.h.
TMVA::DNN::TTensorBatch< Architecture_t >::TTensorBatch | ( | Tensor_t & | inputTensor, |
Matrix_t & | outputMatrix, | ||
Matrix_t & | weightMatrix | ||
) |
Definition at line 198 of file TensorDataLoader.h.
|
default |
|
default |
|
inline |
Return the tensor representing the input data.
Definition at line 77 of file TensorDataLoader.h.
|
inline |
Return the matrix representing the output data.
Definition at line 79 of file TensorDataLoader.h.
|
inline |
Return the matrix holding the event weights.
Definition at line 81 of file TensorDataLoader.h.
|
default |
|
default |
|
private |
The input tensor batch, one matrix one input.
Definition at line 65 of file TensorDataLoader.h.
|
private |
The output matrix representing the ground truth.
Definition at line 66 of file TensorDataLoader.h.
|
private |
The event/example weights.
Definition at line 67 of file TensorDataLoader.h.