How can I apply a linear transformation on sparse matrix in PyTorch?
In PyTorch, we have nn.linear
that applies a linear transformation to the incoming data:
y = WA+b
In this formula, W
and b
are our learnable parameters and A
is my input data matrix. The matrix 'A' for my case is too large for RAM to complete loading, so I use it sparsely. Is it possible to perform such an operation on sparse matrices using PyTorch?
Comments
Post a Comment