nottombrown/parent_of_node.py

Created April 24, 2018 01:21

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/nottombrown/c95160bc1a8ba0e4668018308c5818a3.js"></script>
Save nottombrown/c95160bc1a8ba0e4668018308c5818a3 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

parent_of_node.py

[tensor for tensor in tensors if preds_node.op in tensor.consumers()]

Author

nottombrown commented Apr 24, 2018

This is how you get inputs to a node

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment