Pytorch transform api

[AbhinavTuli]

🚀 🚀 Pull Request

Checklist:

• My code follows the style guidelines of this project and the Contributing document
• I have commented my code, particularly in hard-to-understand areas
• I have kept the coverage-rate up
• I have performed a self-review of my own code and resolved any problems
• I have checked to ensure there aren't any other open Pull Requests for the same change
• I have described and made corresponding changes to the relevant documentation
• New and existing unit tests pass locally with my changes

Changes

Allows another way to specify pytorch transforms using a dictionary instead of a function that is applied on the entire resultant dictionary of data returned by pytorch. The old way is still supported.

For example:-

# ds has image, label and mask as 3 tensors
ptds = ds.pytorch(transform = {"image" : rotate, "label": None})

# this will transform images, return labels as it is and not return masks. 

[CLAassistant]

All committers have signed the CLA.

[codecov]

Codecov Report

Merging #1337 (a1888de) into main (d43ec3c) will increase coverage by 0.72%.
The diff coverage is 98.59%.

@@            Coverage Diff             @@
##             main    #1337      +/-   ##
==========================================
+ Coverage   90.07%   90.80%   +0.72%     
==========================================
  Files         146      146              
  Lines       10326    10533     +207     
==========================================
+ Hits         9301     9564     +263     
+ Misses       1025      969      -56     

Flag	Coverage Δ
unittests	90.80% <98.59%> (+0.72%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
hub/integrations/pytorch/common.py	85.71% <96.55%> (+24.17%)	⬆️
hub/core/io.py	97.25% <100.00%> (-0.12%)	⬇️
hub/integrations/pytorch/dataset.py	91.49% <100.00%> (ø)
hub/integrations/pytorch/pytorch.py	86.20% <100.00%> (+2.20%)	⬆️
hub/integrations/tests/test_pytorch.py	100.00% <100.00%> (ø)
hub/integrations/tests/test_pytorch_dataloader.py	95.69% <100.00%> (+0.76%)	⬆️
hub/core/lock.py	82.22% <0.00%> (-8.89%)	⬇️
hub/cli/commands.py	100.00% <0.00%> (ø)
hub/api/tests/test_json.py	100.00% <0.00%> (ø)
hub/api/tests/test_text.py	100.00% <0.00%> (ø)
... and 36 more

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update d43ec3c...a1888de. Read the comment docs.

[alphazero]

@AbhinavTuli @farizrahman4u @aliubimov Going forward please prefix all PRs and related commit logs with Jira [ticket-id].

[alphazero]

It's a bit tedious but better practice to not hardcode values.