fastNLP.core.collate_fn module

undocumented

class fastNLP.core.collate_fn.ConcatCollateFn(inputs, output, pad_val=0, max_len=0, is_input=True, is_target=False)[源代码]

基类:object

别名 fastNLP.ConcatCollateFn fastNLP.core.ConcatCollateFn

field拼接collate_fn,将不同field按序拼接后,padding产生数据。

参数
  • inputs (List[str]) – 将哪些field的数据拼接起来, 目前仅支持1d的field

  • output (str) – 拼接后的field名称

  • pad_val – padding的数值

  • max_len – 拼接后最大长度

  • is_input – 是否将生成的output设置为input

  • is_target – 是否将生成的output设置为target