dev_AI_framework

layers - activation

명징직조지훈 2024. 8. 26. 19:44

activation class

from dev.layers.layer import Layer
from dev.activations import activations

class Activation(Layer):
    def __init__(self, activation, **kwargs):
        super().__init__(**kwargs)
        self.activation = activations.get(activation)

    def call(self, inputs):
        return self.activation(inputs)
    
    def compute_output_shape(self, input_shape):
        return input_shape

activation 클래스 구성 요소, 메서드