diff --git a/src/main/java/org/dromara/easyai/transFormer/model/MultiSelfAttentionModel.java b/src/main/java/org/dromara/easyai/transFormer/model/MultiSelfAttentionModel.java new file mode 100644 index 0000000..74b90b7 --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/MultiSelfAttentionModel.java @@ -0,0 +1,33 @@ +package org.dromara.easyai.transFormer.model; + +import java.util.List; + +public class MultiSelfAttentionModel { + private float[][] powerModel; + private List qkvModelList; + private int depth; + + public int getDepth() { + return depth; + } + + public void setDepth(int depth) { + this.depth = depth; + } + + public float[][] getPowerModel() { + return powerModel; + } + + public void setPowerModel(float[][] powerModel) { + this.powerModel = powerModel; + } + + public List getQkvModelList() { + return qkvModelList; + } + + public void setQkvModelList(List qkvModelList) { + this.qkvModelList = qkvModelList; + } +}