Hi,
I use type annotation for my code.
Is there any typing annotation guideline for pytorch?
I want to do something like this.
class MyModule(nn.Module):
  def __init__(self):
    super().__init__()
    self.linear = nn.Linear(10, 4)

  def forward(self, x: torch.Tensor[torch.float, [B, 10]]) -> torch.Tensor[torch.float, [B, 4]]:
    return self.linear(x)

I guess torchScript or some other approach, e.g. View(?), could support this one, but I couldn’t find official or near-official document.
Is there anyone who know about this, help me 
Thanks,