!2255 Fix doxygen comments for bucket batch op.

Merge pull request !2255 from Peilin/bucket-batch-doxygen-fix
pull/2255/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e94ad5241b

@ -187,7 +187,7 @@ class Dataset:
bucket for [bucket_boundaries[i], bucket_boundaries[i+1]) for each
0<i<n, and one bucket for [bucket_boundaries[n-1], inf).
bucket_batch_sizes (list of int): A list consisting of the batch sizes for
each buclet. Must contain len(bucket_boundaries)+1 elements.
each bucket. Must contain len(bucket_boundaries)+1 elements.
element_length_function (Callable, optional): A function that takes in
len(column_names) arguments and returns an int. If no value is
provided, then len(column_names) must be 1, and the size of the first
@ -228,7 +228,7 @@ class Dataset:
>>>
>>> data = data.bucket_batch_by_length(column_names, bucket_boundaries,
>>> bucket_batch_sizes,
>>> element_length_function, pad_info),
>>> element_length_function, pad_info,
>>> pad_to_bucket_boundary)
"""
return BucketBatchByLengthDataset(self, column_names, bucket_boundaries, bucket_batch_sizes,

Loading…
Cancel
Save