def freeze
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackwardimport Base
1115# node wrapped in Frozen{id, .} for each operand id that is a constant of the
1116# newest-first list of count nodes, the first operand outermost.
1117def freeze(ids: List<&2, Nat>, +nodes: List<&2, Node>, +count: Nat, node: Node) -> Node:
1118 match ids:
1119 case Nil{}:
1120 node
1121 case Con{+id, rest}:
1122 frozen_if(id, freeze(rest, nodes, count, node), constant_at(nodes, count, id, at_head(count, id)))