deeplima::graph_dp::train::BatchIterator::Batch Class Reference

#include </home/runner/work/lima/lima/deeplima/libs/tasks/graph_dp/model/iterable_dataset.h>

Public Member Functions
bool	empty () const
size_t	get_batch_size () const
	Batch ()
	Batch (const torch::Tensor &trainable_input, const torch::Tensor &frozen_input, const torch::Tensor &gold, size_t batch_dim=0)
const torch::Tensor &	trainable_input () const
const torch::Tensor &	frozen_input () const
const torch::Tensor &	gold () const

Protected Attributes
const size_t	m_batch_dim = 0
const torch::Tensor	m_trainable_input
const torch::Tensor	m_frozen_input
const torch::Tensor	m_gold

Detailed Description

Definition at line 21 of file iterable_dataset.h.

Constructor & Destructor Documentation

Batch() [1/2]

deeplima::graph_dp::train::BatchIterator::Batch::Batch ( )

inline

Definition at line 34 of file iterable_dataset.h.

Batch() [2/2]

deeplima::graph_dp::train::BatchIterator::Batch::Batch ( const torch::Tensor &  trainable_input, const torch::Tensor &  frozen_input, const torch::Tensor &  gold, size_t  batch_dim = 0  )

inline

Definition at line 36 of file iterable_dataset.h.

Member Function Documentation

empty()

bool deeplima::graph_dp::train::BatchIterator::Batch::empty ( ) const

inline

Definition at line 24 of file iterable_dataset.h.

frozen_input()

const torch::Tensor & deeplima::graph_dp::train::BatchIterator::Batch::frozen_input ( ) const

inline

Definition at line 52 of file iterable_dataset.h.

get_batch_size()

size_t deeplima::graph_dp::train::BatchIterator::Batch::get_batch_size ( ) const

inline

Definition at line 29 of file iterable_dataset.h.

gold()

const torch::Tensor & deeplima::graph_dp::train::BatchIterator::Batch::gold ( ) const

inline

Definition at line 57 of file iterable_dataset.h.

trainable_input()

const torch::Tensor & deeplima::graph_dp::train::BatchIterator::Batch::trainable_input ( ) const

inline

Definition at line 47 of file iterable_dataset.h.

Member Data Documentation

m_batch_dim

const size_t deeplima::graph_dp::train::BatchIterator::Batch::m_batch_dim = 0

protected

Definition at line 63 of file iterable_dataset.h.

m_frozen_input

const torch::Tensor deeplima::graph_dp::train::BatchIterator::Batch::m_frozen_input

protected

Definition at line 66 of file iterable_dataset.h.

m_gold

const torch::Tensor deeplima::graph_dp::train::BatchIterator::Batch::m_gold

protected

Definition at line 67 of file iterable_dataset.h.

m_trainable_input

const torch::Tensor deeplima::graph_dp::train::BatchIterator::Batch::m_trainable_input

protected

Definition at line 65 of file iterable_dataset.h.

---

The documentation for this class was generated from the following file:

• deeplima/libs/tasks/graph_dp/model/iterable_dataset.h