class_docstr: Thresholded Linear Unit, c.f. https://arxiv.org/pdf/1911.09737.pdf .
docstr: Executes this layer as part of a forward pass through the model.
  
  Args:
    inputs: Tensor.
  
  Returns:
    Tensor of same shape and dtype as the input.
docstr_tok: ["Executes","this","layer","as","part","of","a","forward","pass","through","the","model",".","Args",":","inputs",":","Tensor",".","Returns",":","Tensor","of","same","shape","and","dtype","as","the","input","."]
code: def forward(self, inputs):
      """Executes this layer as part of a forward pass through the model.
  
      Args:
        inputs: Tensor.
  
      Returns:
        Tensor of same shape and dtype as the input.
      """
      threshold = self.weights
      return jnp.maximum(inputs, threshold)
  
imports: from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import base
  from trax.layers.assert_shape import assert_shape
  from trax.layers.base import Fn
label_desc: Use image node_id 2 for calling the ThresholdedLinearUnit obj's underlying member method code with example usage: obj.forward(inputs) and returns: jnp