---
title: SplitParam
description: "| 属性 | 类型 | 默认值 | 描述 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendtb/ascendtb_01_0340.html
sourcePath: /source/zh/canncommercial/900/API/ascendtb/ascendtb_01_0340.html
indexId: 3e1887ad38b9de615b537067527aa32fe447e277fdc159202cd3b8e063d0c7e864
---
# SplitParam

| 属性 | 类型 | 默认值 | 描述 |
| --- | --- | --- | --- |
| split\_dim | int | 0 | \- |
| split\_num | int | 2 | \- |
| split\_sizes | List[int] | list() | \- |


#### 约束说明

split_sizes的长度大小需小于DEFAULT_SVECTOR_SIZE，否则会抛出长度溢出异常。


#### 调用示例

```
import torch
import torch_atb  

def split():
    input_npu = torch.randn(6, 6, dtype=torch.float16).npu()
    print("input: ", input_npu)
    split_param = torch_atb.SplitParam(split_dim = 0, split_num = 2)
    split = torch_atb.Operation(split_param)

    def split_run():
        split_outputs = split.forward([input_npu])
        return split_outputs

    outputs = split_run()
    print("outputs: ", outputs)

if __name__ == "__main__":
    split()
```