MCPcopy Create free account
hub / github.com/brightmart/text_classification / test_decoder_single_layer

Function test_decoder_single_layer

a07_Transformer/a2_decoder.py:130–135  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

128decoder,Q, K_s=init()
129
130def test_decoder_single_layer():
131 layer_index=0
132 print("Q.get_shape().as_list():",Q.get_shape().as_list())
133 sequence_length_unfold=Q.get_shape().as_list()[0]
134 mask=tf.ones((sequence_length_unfold)) #,dtype=tf.float32
135 decoder.decoder_single_layer(Q,K_s,layer_index,mask)
136
137def test_decoder():
138 output=decoder.decoder_fn()

Callers

nothing calls this directly

Calls 1

decoder_single_layerMethod · 0.80

Tested by

no test coverage detected