split
- (input, num_or_sections, dim=-1, name=None)[源代码]
- input (Variable) - 输入变量,为数据类型为float32,float64,int32,int64的多维Tensor或者LoDTensor。
- dim (int) - 需要分割的维度。如果dim < 0,划分的维度为rank(input) + dim,数据类型为int32,int64。
- name (str,可选) - 具体用法请参见 ,一般无需设置,默认值为None。
本文档使用 全库网 构建