Skip to main content
. 2022 Aug 17;8(8):e10240. doi: 10.1016/j.heliyon.2022.e10240

Table A.1.

VAE encoder architecture. trial dimension = (64 ∗ 1536); n nodes 1 = product (trial dimension) = 98304; kernel size 1 = (1, floor (2 ∗ trial dimension [1]/96)) = (1, 32); kernel size 2 = (trial dimension [0], 1) = (64, 1); strides = (1, max (floor (trial dimension [1]/96), 4)) = (1, 16).

Layer Type Filters Kernel size Strides Output Connected to
Input 1 - - - (None, 2) -
Embedding 1 - - - (None, 2, 10) Input 1
Flatten 1 - - - (None, 20) Embedding 1
Dense 1 5 - - (None, 5) Flatten 1
Dense 2 n nodes 1 - - (None, 98304) Dense 1
Reshape 1 - - - (None, 64, 1536, 1) Dense 2
Input 2 - - - (None, 64, 1536, 1) -
Concatenate 1 - - - (None, 64, 1536, 2) Input 2
Reshape 1
Conv 1 16 kernel size 1 strides (None, 64, 95, 16) Concatenate 1
LeakyReLU 1 - - - (None, 64, 95, 16) Conv 1
Conv 2 32 kernel size 2 - (None, 1, 95, 32) LeakyReLU 1
LeakyReLU 2 - - - (None, 1, 95, 32) Conv 2
Flatten 2 - - - (None, 3040) LeakyReLU 2
Dense 3 16 - - (None, 16) Flatten 2
Dense 4 10 - - (None, 10) Dense 3
Dense 5 10 - - (None, 10) Dense 3
Sampling 1 - - - (None, 10) Dense 4
Dense 5