class_docstr: Lazy multihead mask, combines multiple lazy masks one per head.
class_docstr_tok: ["Lazy","multihead","mask",",","combines","multiple","lazy","masks","one","per","head","."]
code: def shape(self) -> Tuple[int, ...]:
      return (len(self.masks),) + self.masks[0].shape
  
imports: from __future__ import annotations
  import dataclasses
  from typing import Any, Callable, Sequence, Tuple
  import numpy
label_desc: Use image node_id 2 for calling the MultiHeadMask obj's underlying member method code with example usage: obj.shape() and returns: unknown