code: def _nonlinearity(x):
      # swish
      return x * torch.sigmoid(x)
  
imports: import logging
  from typing import Optional, Sequence
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from torch import Tensor
  from neuralcompression.layers import XCA, ChannelNorm2D
  from .._outputs._vqvae_autoencoder_output import VqVaeAutoencoderOutput
label_desc: Use image node_id 1 for calling a global function with example usage: _nonlinearity(x) and returns: unknown