I have problems while running the code for Stochastic steady-state embedding (SSE).
the error is that-AttributeError: ‘NDArray’ object has no attribute ‘device’
I know that this is because I am using mxnet rather than pytorch. But how can I make it suitable for mxnet tensor?