public final class Blocks
extends java.lang.Object
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. |
public static Block batchFlattenBlock()
Block
whose forward function applies the batchFlatten
method.Block
whose forward function applies the batchFlatten
methodpublic static Block batchFlattenBlock(long size)
Block
whose forward function applies the batchFlatten
method. The size of input to the block returned must be batch_size * size.size
- the expected size of each inputBlock
whose forward function applies the batchFlatten
methodpublic static Block batchFlattenBlock(long batch, long size)
Block
whose forward function applies the batchFlatten
method. The size of input to the block returned must be batch * size.batch
- the batch sizesize
- the expected size of each inputBlock
whose forward function applies the batchFlatten
methodpublic static Block identityBlock()
LambdaBlock
that performs the identity function.Block