Module Import alias identifier ImportFrom alias Raise Call NameLoad Str FunctionDef arguments args NameParam defaults body Expr Str Raise Call NameLoad Str decorator_list FunctionDef arguments args NameParam NameParam defaults body Expr Str Raise Call NameLoad Str decorator_list FunctionDef arguments args NameParam defaults body Expr Str Raise Call NameLoad Str decorator_list If CompareEq NameLoad Str body Assign NameStore Call NameLoad Num Assign NameStore Call NameLoad Num Assign NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad ListLoad NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr ListLoad Num Num Num Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr Num Num Num attr AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr Num Num Num Num attr AttributeLoad NameLoad attr Assign NameStore BinOpAdd SubscriptLoad Call AttributeLoad NameLoad attr NameLoad TupleLoad Num Num Num ExtSlice Slice Index NameLoad Slice Slice NameLoad Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr keyword Num attr keyword Num Assign NameStore Call NameLoad NameLoad NameLoad Assert Call AttributeLoad NameLoad attr NameLoad NameLoad TupleLoad NameLoad NameLoad Print Str
N/A N/A numpy np theano function N/A N/A NotImplementedError TODO: add any other imports you need make_tensor N/A N/A dim N/A N/A N/A 
    Returns a new Theano tensor with no broadcastable dimensions.
    dim: the total number of dimensions of the tensor.
    (You can use any dtype you like)
     N/A N/A NotImplementedError TODO: implement this function. N/A broadcasted_add N/A N/A a b N/A N/A N/A 
    a: a 3D theano tensor
    b: a 4D theano tensor
    Returns c, a 4D theano tensor, where

    c[i, j, k, l] = a[l, k, i] + b[i, j, k, l]

    for all i, j, k, l
     N/A N/A NotImplementedError TODO: implement this function. N/A partial_max N/A N/A a N/A N/A N/A 
    a: a 4D theano tensor

    Returns b, a theano matrix, where

    b[i, j] = max_{k,l} a[i, k, l, j]

    for all i, j
     N/A N/A NotImplementedError TODO: implement this function. N/A N/A N/A __name__ __main__ N/A N/A a N/A make_tensor 3 N/A b N/A make_tensor 4 N/A c N/A broadcasted_add a b N/A d N/A partial_max c N/A f N/A function N/A a b d N/A rng N/A N/A N/A np random RandomState N/A 1 2 3 N/A a_value N/A N/A N/A N/A rng randn 2 2 2 astype N/A a dtype N/A b_value N/A N/A N/A N/A rng rand 2 2 2 2 astype N/A b dtype N/A c_value N/A N/A N/A N/A np transpose a_value N/A 2 1 0 N/A N/A N/A None N/A N/A b_value N/A expected N/A N/A N/A N/A c_value max axis 1 max axis 1 N/A actual N/A f a_value b_value N/A N/A N/A np allclose actual expected N/A actual expected N/A SUCCESS!