class_docstr: A layer which fuses the computation of Dropout(Activation(x))
  in a single GPU kernel
class_docstr_tok: ["A","layer","which","fuses","the","computation","of","Dropout","(","Activation","(","x",")",")","in","a","single","GPU","kernel"]
code: def init_weights(self, *args, **kwargs):
      with torch.no_grad():
          if self.bias is not None:
              self.bias.fill_(0.0)
  
file_path: xformers/xformers/triton/dropout.py
imports: from typing import Optional
  import torch
  import triton
  from torch.cuda.amp import custom_bwd, custom_fwd
  from xformers.components.activations import Activation, build_activation
  from xformers.triton.k_activations import get_triton_activation_index
  from xformers.triton.k_dropout import k_dropout_bw, k_dropout_fw
label_desc: Use image node_id 2 for calling the FusedDropoutBias obj's underlying member method code with example usage: obj.init_weights() without return types