(shape, dtype, ctx, low, high)
| 365 | |
| 366 | |
| 367 | def randint(shape, dtype, ctx, low, high): |
| 368 | with tf.device(ctx): |
| 369 | t = tf.random.uniform(shape, dtype=dtype, minval=low, maxval=high) |
| 370 | return t |
| 371 | |
| 372 | |
| 373 | def pad_packed_tensor(input, lengths, value, l_min=None): |