ai.djl.translate

Class PaddingStackBatchifier

java.lang.Object

ai.djl.translate.PaddingStackBatchifier

All Implemented Interfaces:

Batchifier

public final class PaddingStackBatchifier
extends java.lang.Object
implements Batchifier

The padding stack batchifier is a StackBatchifier that also pads elements to reach the same length.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	PaddingStackBatchifier.Builder Builder to build a PaddingStackBatchifier.

Field Summary

Fields inherited from interface ai.djl.translate.Batchifier

STACK

Method Summary

Modifier and Type	Method and Description
NDList	batchify(NDList[] inputs) Converts an array of NDList into an NDList.
static PaddingStackBatchifier.Builder	builder() Returns a PaddingStackBatchifier.Builder.
NDList[]	split(NDList list, int numOfSlices, boolean evenSplit) Splits the given NDList into the given number of slices.
NDList[]	unbatchify(NDList inputs) Reverses the batchify operation.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.Batchifier

fromString

Method Detail

batchify

public NDList batchify(NDList[] inputs)

Converts an array of NDList into an NDList.

The size of the input array is the batch size. The data in each of the NDList are assumed to be the same, and are batched together to form one batched NDList.

Specified by:

batchify in interface Batchifier

Parameters:

inputs - the input array of NDList where each element is a

Returns:

the batchified NDList

unbatchify

public NDList[] unbatchify(NDList inputs)

Reverses the batchify operation.

Specified by:

unbatchify in interface Batchifier

Parameters:

inputs - the NDList that needs to be 'unbatchified'

Returns:

an array of NDLists, of size equal to batch size, where each NDList is one element from the batch of inputs

split

public NDList[] split(NDList list,
                      int numOfSlices,
                      boolean evenSplit)

Splits the given NDList into the given number of slices.

This function unbatchifies the input NDList, redistributes them into the given number of slices, and then batchify each of the slices to form an array of NDList.

Specified by:

split in interface Batchifier

Parameters:

list - the NDList that needs to be split

numOfSlices - the number of slices the list must be sliced into

evenSplit - whether each slice must have the same shape

Returns:

an array of NDList that contains all the slices

builder

public static PaddingStackBatchifier.Builder builder()

Returns a PaddingStackBatchifier.Builder.

Returns:

a PaddingStackBatchifier.Builder