def get_input_feat(self, obs):
    input_a_t = np.zeros((len(obs), args.angle_feat_size), np.float32)
    for (i, ob) in enumerate(obs):
        input_a_t[i] = utils.angle_feature(ob['heading'], ob['elevation'])
    input_a_t = torch.from_numpy(input_a_t).cuda()
    f_t = self._feature_variable(obs)
    (candidate_feat, candidate_leng) = self._candidate_variable(obs)
    return (input_a_t, f_t, candidate_feat, candidate_leng)
def get_input_feat(self, obs):
    input_a_t = np.zeros((len(obs), args.angle_feat_size), np.float32)
    for (i, ob) in enumerate(obs):
        input_a_t[i] = utils.angle_feature(ob['heading'], ob['elevation'])
    input_a_t = torch.from_numpy(input_a_t).cuda()
    f_t = self._feature_variable(obs)
    (candidate_feat, candidate_leng) = self._candidate_variable(obs)
    return (input_a_t, f_t, candidate_feat, candidate_leng)