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'.
--
https://mail.python.org/mailman/listinfo/python-list

Reply via email to