Module Import alias identifier FunctionDef arguments args NameParam NameParam defaults body Expr Str Return Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr Num NameLoad decorator_list FunctionDef arguments args NameParam defaults body Expr Str Assign NameStore Call AttributeLoad NameLoad attr BinOpSub NameLoad Call AttributeLoad NameLoad attr NameLoad Return BinOpDiv NameLoad Call AttributeLoad NameLoad attr NameLoad decorator_list
N/A N/A numpy np categorical N/A N/A p rng N/A N/A N/A 
    Draws multinomial samples from distribution p
     N/A N/A N/A np argmax N/A N/A rng multinomial 1 p N/A softmax N/A N/A z N/A N/A N/A 
    Computes softmax values for each Q-value in x
     N/A ex N/A N/A np exp N/A z N/A N/A np max z N/A N/A ex N/A N/A np sum ex N/A