utils – Utilities functions operating on the graph

Reference

class theano.gof.utils.AssocList[source]

An associative list.

This class is like a dict that accepts unhashable keys by using an assoc list for internal use only

exception theano.gof.utils.MethodNotDefined[source]

To be raised by functions defined as part of an interface.

When the user sees such an error, it is because an important interface function has been left out of an implementation class.

exception theano.gof.utils.TestValueError[source]

Base exception class for all test value errors.

class theano.gof.utils.ValidatingScratchpad(attr, attr_filter)[source]

This Scratchpad validates attribute values.

theano.gof.utils.add_tag_trace(thing, user_line=None)[source]

Add tag.trace to an node or variable.

The argument is returned after being affected (inplace).

Parameters
  • thing – The object where we add .tag.trace.

  • user_line – The max number of user line to keep.

Notes

We also use config.traceback__limit for the maximum number of stack level we look.

theano.gof.utils.difference(seq1, seq2)[source]

Returns all elements in seq1 which are not in seq2: i.e seq1\seq2.

theano.gof.utils.flatten(a)[source]

Recursively flatten tuple, list and set in a list.

theano.gof.utils.hash_from_file(file_path)[source]

Return the SHA256 hash of a file.

theano.gof.utils.memoize(f)[source]

Cache the return value for each tuple of arguments (which must be hashable).

theano.gof.utils.remove(predicate, coll)[source]

Return those items of collection for which predicate(item) is true.

Examples

>>> def even(x):
...     return x % 2 == 0
>>> remove(even, [1, 2, 3, 4])
[1, 3]
theano.gof.utils.simple_extract_stack(f=None, limit=None, skips=None)[source]

This is traceback.extract_stack from python 2.7 with this change:

  • Comment the update of the cache.

  • Skip internal stack trace level.

The update of the cache call os.stat to verify is the cache is up to date. This take too much time on cluster.

limit - The number of stack level we want to return. If None, mean all what we can.

skips - partial path of stack level we don’t want to keep and count.

When we find one level that isn’t skipped, we stop skipping.

theano.gof.utils.toposort(prereqs_d)[source]

Sorts prereqs_d.keys() topologically.

prereqs_d[x] contains all the elements that must come before x in the ordering.

theano.gof.utils.uniq(seq)[source]

Do not use set, this must always return the same value at the same index. If we just exchange other values, but keep the same pattern of duplication, we must keep the same order.