提交 849f5268 编写于 作者: A A. Unique TensorFlower

Internal change

PiperOrigin-RevId: 524906464
上级 e01cbc24
......@@ -37,7 +37,7 @@ def sample_random_sequence(model_input, num_frames, num_samples):
max_start_frame_index = tf.maximum(num_frames - num_samples, 0)
start_frame_index = tf.cast(
tf.multiply(
tf.random_uniform([batch_size, 1]),
tf.random.uniform([batch_size, 1]),
tf.cast(max_start_frame_index + 1, tf.float32)), tf.int32)
frame_index = tf.minimum(start_frame_index + frame_index_offset,
tf.cast(num_frames - 1, tf.int32))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册