ai.djl.nn

Class Blocks

java.lang.Object

ai.djl.nn.Blocks

public final class Blocks
extends java.lang.Object

Utility class that provides some useful blocks.

Method Summary

Modifier and Type	Method and Description
static Block	batchFlattenBlock() Creates a Block whose forward function applies the batchFlatten method.
static Block	batchFlattenBlock(long size) Creates a Block whose forward function applies the batchFlatten method.
static Block	batchFlattenBlock(long batch, long size) Creates a Block whose forward function applies the batchFlatten method.
static Block	identityBlock() Creates a LambdaBlock that performs the identity function.

Methods inherited from class java.lang.Object

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

Method Detail

batchFlattenBlock

public static Block batchFlattenBlock()

Creates a Block whose forward function applies the batchFlatten method.

Returns:

a Block whose forward function applies the batchFlatten method

batchFlattenBlock

public static Block batchFlattenBlock(long size)

Creates a Block whose forward function applies the batchFlatten method. The size of input to the block returned must be batch_size * size.

Parameters:

size - the expected size of each input

Returns:

a Block whose forward function applies the batchFlatten method

batchFlattenBlock

public static Block batchFlattenBlock(long batch,
                                      long size)

Creates a Block whose forward function applies the batchFlatten method. The size of input to the block returned must be batch * size.

Parameters:

batch - the batch size

size - the expected size of each input

Returns:

a Block whose forward function applies the batchFlatten method

identityBlock

public static Block identityBlock()

Creates a LambdaBlock that performs the identity function.

Returns:

an identity Block