RuntimeError: stft input and window must be on the same device but got self on cpu and window on cuda:0
时间: 2024-05-24 18:13:56 浏览: 363
这个错误的含义是:STFT输入和窗口必须在同一个设备上,但是你给出的STFT输入数据在CPU上,而窗口数据在CUDA设备上。解决这个问题需要将输入数据移动到与窗口数据相同的设备上。可以使用以下代码来将CPU上的数据移动到CUDA设备上:input_data = input_data.to("cuda:0")。如果你的代码中还涉及到其他类型的数据转移,也可以使用to方法来将数据移动到需要的设备上。
阅读全文