Interface ActivationVectorizationProvider

public interface ActivationVectorizationProvider

Method Summary

Modifier and Type	Method	Description
void	applySoftmaxBatchVectorized(Tensor2D logits)
void	applySoftmaxVectorized(Tensor1D outputs1D, float maxWs)
void	leakyRelu(Tensor tensor, float a)
void	relu(Tensor tensor)
void	sigmoid(Tensor tensor)
void	tanh(Tensor tensor)

Method Details

relu

void relu(Tensor tensor)

tanh

void tanh(Tensor tensor)

sigmoid

void sigmoid(Tensor tensor)

leakyRelu

void leakyRelu(Tensor tensor,
 float a)

applySoftmaxVectorized

void applySoftmaxVectorized(Tensor1D outputs1D,
 float maxWs)

applySoftmaxBatchVectorized

void applySoftmaxBatchVectorized(Tensor2D logits)