input,  Placeholder:0, TF_FLOAT, [2, 3, 4]
output, Softmax:0,     TF_FLOAT, [2, 3, 4]