diff --git a/fastNLP/core/collators/padders/oneflow_padder.py b/fastNLP/core/collators/padders/oneflow_padder.py index 30d73e26..3f2b8bce 100644 --- a/fastNLP/core/collators/padders/oneflow_padder.py +++ b/fastNLP/core/collators/padders/oneflow_padder.py @@ -169,7 +169,7 @@ class OneflowTensorPadder(Padder): else: max_shape = [len(batch_field)] + [max(*_) for _ in zip(*shapes)] - tensor = oneflow.full(max_shape, value=pad_val, dtype=dtype, device=device) + tensor = oneflow.full(max_shape, fill_value=pad_val, dtype=dtype, device=device) for i, field in enumerate(batch_field): slices = (i, ) + tuple(slice(0, s) for s in shapes[i]) tensor[slices] = field @@ -221,6 +221,6 @@ def get_padded_oneflow_tensor(batch_field, dtype=None, pad_val=0): :return: """ shapes = get_shape(batch_field) - tensor = oneflow.full(shapes, dtype=dtype, value=pad_val) + tensor = oneflow.full(shapes, dtype=dtype, fill_value=pad_val) tensor = fill_tensor(batch_field, tensor, dtype=dtype) return tensor