snapshot...

modules/deeplearning/icing_fcn.py

@@ -400,6 +400,8 @@ class IcingIntensityFCN:
         nda = np.zeros([1])
         nda[0] = self.flight_level
         nda = tf.one_hot(nda, 5).numpy()
+        nda = np.expand_dims(nda, axis=0)
+        nda = np.expand_dims(nda, axis=0)
 
         return data, nda