[QUESTION] torch.as_tensor(nvcv.Tensor.cuda()) copy the data from gpu to cpu?

[QinlongHuang]

What is your question?
Great work on acceleration on CV tasks! But I find when I convert a nvcv.Tensor to torch.Tensor, it will copy the data in GPU to CPU.
Is it a bug? Or any memory consideration?

a = torch.randn(10, 256, 256, 3).to('cuda')  # GPU
a_n = nvcv.as_tensor(a, 'NHWC')  # GPU
b = torch.as_tensor(a_n.cuda())  # CPU

Env Info:
RTX 4090, PyTorch 2.0.1

[QinlongHuang]

Another question, can I convert ImageBatchVatShape to nvcv.Tensor or torch.Tensor directly? As far as I know, I have to process it one by one.

[lokvke]

Another question, can I convert ImageBatchVatShape to nvcv.Tensor or torch.Tensor directly? As far as I know, I have to process it one by one.

+1