Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def f(n):
- z = []
- # We ask you to tell us the element dtype of the list
- autograph.set_element_type(z, tf.int32)
- for i in range(n):
- z.append(i)
- # when you're done with the list, stack it
- # (this is just like np.stack)
- return autograph.stack(z)
Add Comment
Please, Sign In to add comment