Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::data::samplers::BatchSize Struct Reference

A wrapper around a batch size value, which implements the CustomBatchRequest interface. More...

#include <stream.h>

Inheritance diagram for torch::data::samplers::BatchSize:
torch::data::samplers::CustomBatchRequest

Public Member Functions

 BatchSize (size_t size)
 
size_t size () const noexceptoverride
 The number of elements accessed by this index.
 
 operator size_t () const noexcept
 

Data Fields

size_t size_
 

Detailed Description

A wrapper around a batch size value, which implements the CustomBatchRequest interface.

Definition at line 23 of file stream.h.


The documentation for this struct was generated from the following files: