class_docstr: Stores inputs to the `memory_efficient_attention` operators
code: def scale_float(self) -> float:
      return (
          self.query.shape[-1] ** (-0.5)
          if self.scale is None
          else self.scale
      )
  
imports: import math
  from dataclasses import dataclass
  from functools import partial
  from typing import Any, Callable, List, Mapping, Optional, Set, Tuple, Type, Union
  import torch
  from ..._cpp_lib import _built_with_cuda
  from ..common import BaseOperator
  from .attn_bias import AttentionBias, BlockDiagonalMask, LowerTriangularMask, LowerTriangularMaskWithTensorBias
label_desc: Use image node_id 2 for calling the Inputs obj's underlying member method code with example usage: obj.scale_float() and returns: unknown