Add support for torch.cat([...], dim=0) in sparse Jacobian

Add support for torch.cat([...], dim=0) in sparse Jacobian #35