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

| 属性 | 类型 | 默认值 | 描述 |
| --- | --- | --- | --- |
| rotary\_coeff | int | 4 | \- |
| cos\_format | int | 0 | \- |


#### 调用示例

```
import torch
import torch_atb

def rope():
    rope_param = torch_atb.RopeParam(rotary_coeff = 4)
    rope = torch_atb.Operation(rope_param)

    def gen_inputs():
        ntoken = 4
        seqlen = 4
        hidden_size = 4096
        head_size = 128
        intensor0 = torch.rand(ntoken, hidden_size).half()
        intensor1 = torch.rand(ntoken, hidden_size).half()
        intensor2 = torch.rand(ntoken, head_size).half()
        intensor3 = torch.rand(ntoken, head_size).half()
        intensor4 = torch.tensor([seqlen], dtype=torch.int32)
        return [intensor0, intensor1, intensor2, intensor3, intensor4]

    in_tensors = gen_inputs()
    in_tensors_npu = [tensor.npu() for tensor in in_tensors]
    print("in_tensors: ", in_tensors_npu)

    def rope_run():
        rope_outputs = rope.forward(in_tensors_npu)
        return rope_outputs

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

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