shapels

Shape inference for PyTorch tensors directly inside VS Code. The extension bundles the shapels language server binary for Linux, macOS and Windows so users can get diagnostics without extra setup.

Features

• Reports tensor shape inference diagnostics as you type.
• Runs inference for tensor shapes, allowing for hovering over variables to inspect their shape.

Requirements

No external dependencies. The platform-specific server binary is shipped inside the extension (shapels-bin/<platform>/shapels).

Extension Settings

• shapels.serverPath: Override the bundled server binary path.

Release Notes

0.2.0

• Support for new 57 shape-wise NoOps like softmax, contiguous or relu.
• Fixed bug on unsqueeze(0).
• Fixed bug on non-recursive inference over view/reshape(EXPR)