forward method

Tensor forward(
  1. Tensor x,
  2. List<Tensor> tracker
)

Implementation

Tensor forward(Tensor x, List<Tensor> tracker) {
  final out = Tensor.layerNorm(x, gamma, beta, eps);
  tracker.add(out);
  return out;
}