Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfl.model_info.LinearNode

View source on GitHub

Represents a linear layer.

@staticmethod
tfl.model_info.LinearNode(
    _cls, input_nodes, coefficients, bias
)

Attributes:

  • input_nodes: List of input nodes to the linear layer.
  • coefficients: Linear weights.
  • bias: Bias term for the linear layer.