X as sparse tensor in Data

[niknoproblems]

Is it possible to use sparse tensor for X?

[rusty1s]

It depends. In general yes. However, you might need to modify mini-batch mode behavior since SparseTensors are generally stacked diagonally instead of row-wise in DataLoader.

[A11en0]

How to modify it? could you please give an example?

[rusty1s]

class MyData(Data):
    def __cat_dim__(self, key, value):
        if key == 'x':
            return 0
        return super().__cat_dim__(key, value)

[A11en0]

thanks!

[rusty1s]

Is your last comment resolved? I cannot find it anymore :)

[A11en0]

yes, I solved it! Thanks!