Fix error of get_action in script version

[Jasonxu1225]

In the paper, we should use K to get action. So I think the script version is wrong because it use N to get_action instead of K. (Although the paper says that IQN Is not sensitive to the value of K). So I modify the IQN to get_action with K, which corresponding to the paper.