output, out:0,  TF_FLOAT,   [1, 3, 4, 4]