projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 6708532) | patch
[DataPipe] Improve Mapper to accept input/output index when apply fn (#64951)
authorErjia Guan <erjia@fb.com>
Tue, 14 Sep 2021 22:44:57 +0000 (15:44 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 14 Sep 2021 22:46:42 +0000 (15:46 -0700)
commitc65128679bec5c2970838b143f0efee08002ea69
tree16c3ec944ae22b9cfc5dcf6cb23279765bd79613tree | snapshot
parent670853295addc9250d5829a26bd1065ff9bff159commit | diff
[DataPipe] Improve Mapper to accept input/output index when apply fn (#64951)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64951

Test Plan: Imported from OSS

Reviewed By: VitalyFedyunin

Differential Revision: D30910035

Pulled By: ejguan

fbshipit-source-id: d687fe10939920a3617a60552fe743e8526438a0
test/test_datapipe.py diff | blob | history
torch/utils/data/datapipes/iter/callable.py diff | blob | history
Domain: Machine Learning / ML Framework;