This PR adds minor refinements to the type inference engine, including:
Adding type signature for torch.nn.Embedding (which I implemented in attempt to typecheck SNLI example but gave up in the middle because the program creates user-defined object in forward)
Add a function to compute the type of x.shape and x.size (where x is a tensor), which used to be embedded in the inference engine
This PR adds minor refinements to the type inference engine, including:
torch.nn.Embedding
(which I implemented in attempt to typecheck SNLI example but gave up in the middle because the program creates user-defined object inforward
)x.shape
andx.size
(wherex
is a tensor), which used to be embedded in the inference engine