code: def _relu(x: float) -> float:
      if isinstance(x, torch.Tensor):
          return cast(float, nn.functional.relu(x))
      else:
          return np.maximum(x, 0)
  
imports: from __future__ import annotations
  import logging
  from typing import cast
  from typing_extensions import Literal
  import numpy
  import torch
  from torch import nn
  from nni.mutable import Sample
  from nni.nas.profiler import Profiler, ExpressionProfiler
  from .supermodule._expression_utils import expression_expectation
label_desc: Use image node_id 3 for calling a global function with example usage: _relu(x) and returns: cast, np