search

lf1-io / padl

Functional deep learning
Apache License 2.0
106 stars 4 forks source link

Support `torch.device` object with `Transform.pd_to` #437

Open wuhu opened 2 years ago

wuhu commented 2 years ago

🛰️ Feature

Currenly we only support string, which is different form pytorch.