NEW: Wrap lamino operator in Pytorch module - Githubissues

AdvancedPhotonSource / tike

Repository for ptychography software

http://tike.readthedocs.io

Other

29 stars 15 forks source link

NEW: Wrap lamino operator in Pytorch module #166

Open carterbox opened 3 years ago

carterbox commented 3 years ago

Purpose

Allow integrating Laminography operator into Pytorch framework

Approach

Wrap the tike cupy operators in the Pytorch API. This is pretty easy because of pytorch/cupy interop.

Pre-Merge Checklists

Submitter

[x] Write a helpfully descriptive pull request title.
[x] Organize changes into logically grouped commits with descriptive commit messages.
[ ] Document all new functions.
[x] Write tests for new functions or explain why they are not needed.
[ ] Build the documentation successfully
[x] Use yapf to format python code.

Reviewer

[ ] Actually read all of the code.
[ ] Run the new code yourself.
[ ] Write a summary of the changes as you understand them.
[ ] Thank the submitter.