ItalianSC:

thanks tom,you said I can create a cuda tensor and use the tensor.data_ptr() as a destination, so can I be sure the return of datr_ptr() from a cuda tensor is a cuda memory address ?


Yes, this is what PyTorch uses internally, too.