RuntimeError: Could not run 'aten::ne.Tensor' with arguments from the 'SparseCPU' backend.=

用torch_geometric调用GATConv出现下面的报错，如何解决？

RuntimeError: Could not run 'aten::ne.Tensor' with arguments from the 'SparseCPU' backend. 'aten::ne.Tensor' is only available for these backends: [CPU, QuantizedCPU, Named, Autograd, Profiler, Tracer].