Exemplo n.º 1
0
        public void Save(Stream stream)
        {
            m_attentionLayer.Save(stream);
            foreach (LSTMAttentionDecoderCell item in m_decoders)
            {
                item.Save(stream);
            }

            m_decoderFFLayer.Save(stream);
        }
Exemplo n.º 2
0
        public void Save(Stream stream)
        {
            Q.Save(stream);
            K.Save(stream);
            V.Save(stream);


            W0.Save(stream);
            b0.Save(stream);

            layerNorm1.Save(stream);
            layerNorm2.Save(stream);
            feedForwardLayer1.Save(stream);
            feedForwardLayer2.Save(stream);
        }
Exemplo n.º 3
0
        public void Save(Stream stream)
        {
            foreach (MultiHeadAttention item in m_selfAttns)
            {
                item.Save(stream);
            }

            foreach (MultiHeadAttention item in m_encAttns)
            {
                item.Save(stream);
            }

            foreach (var item in m_posFFNs)
            {
                item.Save(stream);
            }


            layerNorm.Save(stream);
            m_decoderFFLayer.Save(stream);
        }
Exemplo n.º 4
0
 public void Save(Stream stream)
 {
     layerNorm2.Save(stream);
     feedForwardLayer1.Save(stream);
     feedForwardLayer2.Save(stream);
 }