Add File

src/main/java/org/dromara/easyai/transFormer/model/FirstDecoderModel.java

@@ -0,0 +1,22 @@
+package org.dromara.easyai.transFormer.model;
+
+public class FirstDecoderModel {
+    private MultiSelfAttentionModel multiSelfAttentionModel;//注意力层model
+    private LayNormModel attentionLayNormModel;//残差1层model
+
+    public MultiSelfAttentionModel getMultiSelfAttentionModel() {
+        return multiSelfAttentionModel;
+    }
+
+    public void setMultiSelfAttentionModel(MultiSelfAttentionModel multiSelfAttentionModel) {
+        this.multiSelfAttentionModel = multiSelfAttentionModel;
+    }
+
+    public LayNormModel getAttentionLayNormModel() {
+        return attentionLayNormModel;
+    }
+
+    public void setAttentionLayNormModel(LayNormModel attentionLayNormModel) {
+        this.attentionLayNormModel = attentionLayNormModel;
+    }
+}