code: def __init__(self, input_dim, output_dim):
      super().__init__()
  
      self.input_dim = input_dim
      self.output_dim = output_dim
      self.linear = nn.Linear(input_dim, output_dim, bias=False)
      with torch.no_grad():
          self.linear.weight.fill_(0.0)
  
imports: from __future__ import annotations
  import pytest
  import nni
  from nni.mutable import *
  from nni.nas.evaluator import FunctionalEvaluator
  from nni.nas.execution import SequentialExecutionEngine
  from nni.nas.space import *
  from nni.nas.strategy import GridSearch, Random, RegularizedEvolution, PolicyBasedRL
  import torch
  from torch import nn
  from tianshou.policy import A2CPolicy
  from tianshou.data import to_torch
label_desc: Use image node_id 1 to create a new ActorNetwork object from inherited base classes: nn with example: obj = ActorNetwork(input_dim, output_dim)