class_docstr: Version using (complex) diagonal state matrix (S4D).
class_docstr_tok: ["Version","using","(","complex",")","diagonal","state","matrix","(","S4D",")","."]
code: def default_state(self, *batch_shape):
      C = _r2c(self.C)
      state = torch.zeros(
          *batch_shape, self.H, self.N, dtype=C.dtype, device=C.device
      )
      return state
  
imports: import logging
  import math
  import os
  from functools import wraps
  from typing import Any, Callable, Optional
  import numpy
  import opt_einsum
  import torch
  import torch.nn
  import torch.nn.functional
  from einops import rearrange, repeat
  from espnet2.asr.state_spaces.components import Activation, DropoutNd, LinearActivation
label_desc: Use image node_id 6 for calling the SSKernelDiag obj's underlying member method code with example usage: obj.default_state() and returns: state