thank you for answer.
There is another difference.
I have a generating function
‘’’
aInput = f()
‘’’
that generates data, and then I have a conversion step,
‘’’
aInput=aInput.cuda().
‘’’
Could that slow it down?