Source code for ztlearn.dl.layers.base

# -*- coding: utf-8 -*-

from abc import ABC, abstractmethod

class Layer(ABC):

    def __init__(self, layer_name = 'zeta_squential'):
        self.layer_name = layer_name

    @property
    def input_shape(self):
        return self.__input_shape

    @input_shape.setter
    def input_shape(self, input_shape):
        self.__input_shape = input_shape

    @property
    def output_shape(self):
        return self.input_shape

    @property
    def layer_parameters(self):
        return 0

    @property
    def layer_name(self):
        return self.__class__.__name__

    @abstractmethod
    def pass_forward(self):
        raise NotImplementedError

    @abstractmethod
    def pass_backward(self):
        raise NotImplementedError