How to make activationFunction for TransformerEncoderBlock constructor ?

[iamuspace]

Hi
I want to make a TransformerEncoderBlock instance.
But I failed because of making Function instance for consturctor of TransformerEncoderBlock.

How to make the Function instance for constructor's activationFunction ?

int embeddingSize = 10; 
int headCount = 5; 
int hiddenSize = 10; 
float dropoutProbability = 0.1f; 
Function<NDList,NDList> activationFunction = Activation.reluBlock(); // compile error 
Block block = new SequentialBlock()	
	.add(Blocks.batchFlattenBlock(inputSize))
	.add(new TransformerEncoderBlock(embeddingSize, headCount, hiddenSize, dropoutProbability, activationFunction)) // how to instanciate activation function ?
	.add(Linear.builder().setUnits(outputSize).build());

[iamuspace]

solved.
I implement Function interface.

import ai.djl.ndarray.NDList;
import ai.djl.nn.Activation;
import java.util.function.Function;

public class SigmoidFunction implements Function<NDList,NDList>
{
@OverRide
public NDList apply(NDList t) {
return Activation.sigmoid(t);
}
}

[iamuspace]

lambda expression is more easy.

Function<NDList,NDList> activationFunction = x -> Activation.sigmoid(x);