pub struct MiniBatch { /* private fields */ }Implementations§
Source§impl MiniBatch
impl MiniBatch
pub fn new(rng: StdRng, size: usize) -> MiniBatch
pub fn batch_size(&self) -> usize
Sourcepub fn next(
&mut self,
loader: &dyn DataLoader,
part: &DataSlice,
) -> Result<(Var, Var), AutoDiffError>
pub fn next( &mut self, loader: &dyn DataLoader, part: &DataSlice, ) -> Result<(Var, Var), AutoDiffError>
Get a random set of samples from the data loader.
Sourcepub fn next_data_slice(
&mut self,
data: &Var,
label: &Var,
) -> Result<(Var, Var), AutoDiffError>
pub fn next_data_slice( &mut self, data: &Var, label: &Var, ) -> Result<(Var, Var), AutoDiffError>
Get a random set of samples given the data and label.
pub fn iter_block<'a>( &self, loader: &'a dyn DataLoader, part: &DataSlice, ) -> Result<BlockIterator<'a>, AutoDiffError>
Auto Trait Implementations§
impl Freeze for MiniBatch
impl RefUnwindSafe for MiniBatch
impl Send for MiniBatch
impl Sync for MiniBatch
impl Unpin for MiniBatch
impl UnwindSafe for MiniBatch
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more