class_docstr: Multi-headed attention.
  
  See "Attention Is All You Need" for more details.
class_docstr_tok: ["Multi-headed","attention",".","See","``","Attention","Is","All","You","Need","''","for","more","details","."]
code: def _set_input_buffer(
      self,
      incremental_state: Dict[str, Dict[str, Optional[Tensor]]],
      buffer: Dict[str, Optional[Tensor]],
  ):
      return self.set_incremental_state(
          incremental_state, "attn_state", buffer
      )
  
imports: import math
  import warnings
  from typing import Dict, Optional, Tuple
  import torch
  import torch.nn.functional
  from torch import Tensor, nn
  from torch.nn import Parameter
label_desc: Use image node_id 8 for calling the MultiheadAttention obj's underlying member method code with example usage: obj._set_input_buffer(incremental_state, buffer) and returns: self