fix(networks): replace Tensor | None with Optional[Tensor] for TorchScript compatibility

fix(networks): replace Tensor | None with Optional[Tensor] for TorchScript compatibility #312