brainpy.base.TensorCollector

class brainpy.base.TensorCollector

A ArrayCollector is a dictionary (name, var) with some additional methods to make manipulation of collections of variables easy. A Collection is ordered by insertion order. It is the object returned by DynamicalSystem.vars() and used as input in many DynamicalSystem instance: optimizers, Jit, etc…

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)
assign(inputs)	Assign data to all values.
clear()
copy()
data()	Get all data in each value.
dict()	Get a dict with the key and the value data.
fromkeys([value])	Create a new dictionary with keys from iterable and values set to value.
get(key[, default])	Return the value for key if key is in the dictionary, else default.
items()
keys()
pop(k[,d])	If key is not found, d is returned if given, otherwise KeyError is raised
popitem()	2-tuple; but raise KeyError if D is empty.
replace(key, new_value)	Replace the original key with the new value.
setdefault(key[, default])	Insert key with a value of default if key is not in the dictionary.
subset(var_type)	Get the subset of the (key, value) pair.
unique()	Get a new type of collector with unique values.
update([E, ]**F)	If E is present and has a .keys() method, then does: for k in E: D[k] = E[k] If E is present and lacks a .keys() method, then does: for k, v in E: D[k] = v In either case, this is followed by: for k in F: D[k] = F[k]
values()