search

i-machine-think / diagNNose

diagNNose is a Python library that facilitates a broad set of tools for analysing hidden activations of neural models.
https://diagnnose.readthedocs.io
MIT License
81 stars 8 forks source link

Incorporate tensor device for ShapleyTensors #66

Closed jumelet closed 3 years ago

jumelet commented 3 years ago

https://github.com/i-machine-think/diagnnose/blob/02b536b53e82b5701490b42063b24c5fb348c2f7/diagnnose/attribute/utils.py#L146