torchdiag
