Keraflow
Deep Learning for Python.
|
Wrapper for implementating simple inline layer. More...
Public Member Functions | |
def | __init__ |
Wrapper for implementating simple inline layer.
(nb_samples, x, y, ...)
output_shape
argument.```python
model.add(Lambda(lambda x, factor: x ** 2)) ```
def keraflow.layers.core.Lambda.__init__ | ( | self, | |
output_fn, | |||
output_shape_fn = None , |
|||
arguments = {} , |
|||
kwargs | |||
) |
output_fn | The function to transform input tensor into output tensor. The function should take 1 position argument (the input tensor) and variable numbers of keyword argument whose value are specified by arguments argument. |
output_shape_fn | tuple, list, or (lambda) function. Expected output shape.
|
arguments | dict. Optional keyword arguments to be passed to output_fn . |
kwargs | see Layer.__init__. |