From 88850a03f3a62d98b0af9780041cb6e73e1ad3fe Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:21 +0800 Subject: [PATCH] Add File --- .../easyai/transFormer/model/QKVModel.java | 40 +++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/transFormer/model/QKVModel.java diff --git a/src/main/java/org/dromara/easyai/transFormer/model/QKVModel.java b/src/main/java/org/dromara/easyai/transFormer/model/QKVModel.java new file mode 100644 index 0000000..1006721 --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/QKVModel.java @@ -0,0 +1,40 @@ +package org.dromara.easyai.transFormer.model; + +public class QKVModel { + private float[][] Q; + private float[][] K; + private float[][] V; + private int selfID; + + public int getSelfID() { + return selfID; + } + + public void setSelfID(int selfID) { + this.selfID = selfID; + } + + public float[][] getQ() { + return Q; + } + + public void setQ(float[][] q) { + Q = q; + } + + public float[][] getK() { + return K; + } + + public void setK(float[][] k) { + K = k; + } + + public float[][] getV() { + return V; + } + + public void setV(float[][] v) { + V = v; + } +}