Yes tensor_data.cpu() is slowing the operation. Do you know how I can convert numpy array without converting cpu() ?
tensor_data.cpu()
cpu()