Notice that a Tensor on CUDA cannot be converted to a numpy array directly. You have to move it to CPU first and then convert to a numpy array.
確定! 回上一頁