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 클래스 구성 요소, 메서드