Revert "assert tensor sizes"

This reverts commit 4d8cd31.

Author: Abdelrahman Ogail

hw2/train_pg_f18.py

@@ -202,8 +202,7 @@ def sample_action(self, policy_parameters):
         if self.discrete:
             sy_logits_na = policy_parameters
             # YOUR_CODE_HERE
-            # _, sy_sampled_ac = tf.nn.top_k(sy_logits_na)
-            sy_sampled_ac = tf.squeeze(tf.multinomial(sy_logits_na, 1), axis=[1])
+            _, sy_sampled_ac = tf.nn.top_k(sy_logits_na)
         else:
             sy_mean, sy_logstd = policy_parameters
             # YOUR_CODE_HERE