Update README.md

README.md

@@ -51,6 +51,12 @@ const model = await AutoModel.from_pretrained(MODEL_ID, {
 > This may not be the best way but the documentation is truly lacking and this does the job :-P
 
 ```typescript
+/**
+ * Minimal attention tensor shape we rely on.
+ * Only `dims` and `data` are used (dims = [B=1, H, T, T]).
+ */
+type XtTensor = { dims: number[]; data: ArrayLike<number | bigint> };
+
 /**
  * Collect attentions across layers from a model.forward(...) output.
  *