test/fastNLP
1
0
Fork 0
mirror of https://gitee.com/fastnlp/fastNLP.git synced 2024-12-01 19:57:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

更新oneflow.full的参数

Browse Source
This commit is contained in:
x54-729 2022-07-17 00:10:05 +08:00
parent 189050d25d
commit cce133d806
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fastNLP/core/collators/padders/oneflow_padder.py
View File

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