class_docstr: Scaled dot-product attention mechanism.
code: def __init__(self, scale=64, att_dropout=0.0):
      super().__init__()
      self.softmax = nn.Softmax(dim=-1)
      self.dropout = nn.Dropout(att_dropout)
      self.scale = scale
  
imports: import os
  import math
  import numpy
  import torch
  from torch import nn
  from towhee.models.layers.layers_with_relprop import Einsum, Linear, Dropout, Softmax
label_desc: Use image node_id 1 to create a new Attention object from inherited base classes: nn with example: obj = Attention(scale, att_dropout)