mapping_networks.runtime.ParameterSpec

class mapping_networks.runtime.ParameterSpec(entries: Iterable[ParameterEntry])

Compile parameter shapes and offsets once, then reuse them cheaply.

__init__(entries: Iterable[ParameterEntry]) → None

Methods

__init__(entries)
count(value)
entry(name)
flatten(tree)	Validate and flatten a named parameter tree in compiled order.
from_module(module)	Build a spec from trainable and frozen parameters in module order.
from_named_tensors(named_tensors)
index(value, [start, [stop]])	Raises ValueError if the value is not present.
unflatten(vector)	Create named tensor views over a one-dimensional generated vector.
validate_tree(tree)	Raise ValueError if names or shapes differ from this spec.

Attributes

names
total_numel