可以支持下返回的是一个在GPU上的torch tensor么？

UsingNet / nvjpeg-python

nvjpeg for python

MIT License

91 stars 24 forks source link

Closed zkwangchao closed 3 years ago

zkwangchao commented 3 years ago

目前支持解码完返回一个numpy array，可以支持返回一个torch的tensor么？谢谢

zeng-qinghui commented 3 years ago

torch中有转换函数把numpy array转换成tensor

torch.from_numpy(a_numpy_array)

zkwangchao commented 3 years ago

再从torch 转的话，需要重新cpu->gpu之间的数据copy，会对预测性能有影响