Closed huchenlei closed 6 months ago
Closes #28.
This PR makes sure that after the decode, the output tensor's dtype/device matches the input pixel tensor's dtype/device.
Closes #28.
This PR makes sure that after the decode, the output tensor's dtype/device matches the input pixel tensor's dtype/device.