class_docstr: MultiScaleAttention
class_docstr_tok: ["MultiScaleAttention"]
code: def save_attn(self, attn):
      self.attn = attn
  
imports: import math
  from functools import partial
  from typing import Tuple, List
  import numpy
  import torch
  from torch import nn
  from torch.nn.init import trunc_normal_
  from einops import rearrange
  from towhee.models.layers.layers_with_relprop import GELU, Linear, Dropout, Conv2d, Conv3d, Softmax, Sigmoid, LayerNorm, Einsum, Add, Clone, MaxPool3d, IndexSelect
label_desc: Use image node_id 3 for calling the MultiScaleAttention obj's underlying member method code with example usage: obj.save_attn(attn) without return types