I want to initialize a dense matrix nn.Parameter(size=(N, N)), and then make back-prop to update it. Each element in this matrix maybe useful thus need to update. How could I efficiently format it and send it into GPU to multiple / add with other torch.sparse.tensor ?