diff --git a/src/main/java/org/dromara/easyai/transFormer/model/LayNormModel.java b/src/main/java/org/dromara/easyai/transFormer/model/LayNormModel.java new file mode 100644 index 0000000..69cdb9d --- /dev/null +++ b/src/main/java/org/dromara/easyai/transFormer/model/LayNormModel.java @@ -0,0 +1,23 @@ +package org.dromara.easyai.transFormer.model; + + +public class LayNormModel { + private float[][] bTa;//模型需要保存 + private float[][] power;//模型需要保存 + + public float[][] getbTa() { + return bTa; + } + + public void setbTa(float[][] bTa) { + this.bTa = bTa; + } + + public float[][] getPower() { + return power; + } + + public void setPower(float[][] power) { + this.power = power; + } +}