Hmmm. Thats kind of mysterious. You are printing the actual concrete cost, and that is being displayed, and that’s included in the timing of .data and .item, and yet they give different times?
Because, in order to print the value, the data needs to have been transferred from the gpu: the sync must have already happened.
cost is a zero dimensional tensor right? like, if you print cost.size(), it’s []?