I want to get the sparse tensor when LongTensor is passed to Embedding Layer.

I want to get adjacency matrix as shape (N,N), and it has value at (i,j) from Embedding Layer.
There are a lot of zeros in the adjacency matrix, so I want to get sparse tensor.

The input is LongTensor which contains indices for embedding as shape has (batchsize, N * N).
In default, embedded value is 0 if index is 0. If you want to change this index, change “padding_idx” any number.
The output is adjacency sparse matrix as shape has (batchsize, N, N).