#for every example in the batch, this takes the first vector and 
#passes it to the classifier linear netowrk for prediction.
clsf_logits = self.classifier(x[:,0,:])