ebrahimHappy

I get error on this simple code

4

this is the code: ``` import torch from pytorch_block_sparse import BlockSparseLinear x = torch.randn(32, 128).to('cuda') y = torch.randn(32, 64).to('cuda') model = torch.nn.Sequential( BlockSparseLinear(128, 64) ).to('cuda') y_pred = model(x) loss =...