grain.transforms.Batch

class grain.transforms.Batch(batch_size: 'int', drop_remainder: 'bool' = False)

Parameters:

• batch_size (int)

• drop_remainder (bool)

__init__(batch_size, drop_remainder=False)

Parameters:

• batch_size (int)

• drop_remainder (bool)

Return type:

None

Methods

__init__(batch_size[, drop_remainder])

Attributes

drop_remainder
batch_size