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

| 属性 | 类型 | 默认值 | 描述 |
| --- | --- | --- | --- |
| activation\_type | torch\_atb.ActivationType | torch\_atb.ActivationType.ACTIVATION\_UNDEFINED | 此默认类型不可用，用户须配置此项参数。 |
| scale | float | 1.0 | \- |
| dim | int | \-1 | \- |
| gelu\_mode | torch\_atb.ActivationParam.GeLUMode | torch\_atb.ActivationParam.GeLUMode.TANH\_MODE | \- |


#### ActivationParam.GeLUMode

枚举项：

- TANH_MODE
- NONE_MODE


#### 调用示例

```
import torch
import torch_atb  

def activation():
    activation_param = torch_atb.ActivationParam(
        activation_type = torch_atb.ActivationType.ACTIVATION_SWISH, scale = 1.0)
    activation = torch_atb.Operation(activation_param)

    intensor = torch.rand(2, 3, 5).bfloat16()
    intensor_npu = intensor.npu()
    print("intensor: ", intensor_npu)

    def activation_run():
        activation_outputs = activation.forward([intensor_npu])
        return activation_outputs

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

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