Class PyTorchEinsumGeneratorContext

Inheritance Relationships

Base Type

Class Documentation

class PyTorchEinsumGeneratorContext : public sequant::PythonEinsumGeneratorContext

Context for PyTorch einsum generator.

Public Functions

inline PyTorchEinsumGeneratorContext()
~PyTorchEinsumGeneratorContext() = default
inline PyTorchEinsumGeneratorContext(ShapeMap index_shapes)