Browse Source

!2255 Fix doxygen comments for bucket batch op.

Merge pull request !2255 from Peilin/bucket-batch-doxygen-fix
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
e94ad5241b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/dataset/engine/datasets.py

+ 2
- 2
mindspore/dataset/engine/datasets.py View File

@@ -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