question on trax

[MRAB]

On 2021-08-17 16:50, joseph pareti wrote:
> In the following code, where does tl.Fn come from? i see it nowhere in the
> documents, i.e I was looking for trax.layers.Fn :
> 
> import numpy as np
> *from trax import layers as tl*
> from trax import shapes
> from trax import fastmath
> #
> def Addition():
>      layer_name = "Addition"  # don't forget to give your custom layer a
> name to identify
> 
>      # Custom function for the custom layer
>      def func(x, y):
>          return x + y
> 
>      return *tl.Fn*(layer_name, func)
> 
[snip]
It comes from using the line:

     from trax import layers as tl

so it's equivalent to 'trax.layers.Fn'.