---
title: LARSV2
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/TensorFlowCommercial/latest/migration/tfmigr1/tfmigr1_tfadapi_0070.html
sourcePath: /source/zh/TensorFlowCommercial/900/migration/tfmigr1/tfmigr1_tfadapi_0070.html
indexId: 3c530fe86c5b5a4741df97067a330ff2be4ccd1045c69457cab85b1e511c963c79
---
# LARSV2

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | ☓ |
| Atlas 推理系列产品 | ☓ |
| Atlas 训练系列产品 | √ |


#### 功能说明

该算子基于权重的范数和梯度的范数在不同层级上使用不同的学习率，对梯度缩放。通常用于提升大batch size场景下的训练精度，用于大规模集群训练，减少训练时间。


#### 函数原型

```
def LARSV2(input_weight,
           input_grad,
           weight_decay,
           learning_rate,
           hyperpara=0.001,
           epsilon=0.00001,
           use_clip=False,
           name=None)
```


#### 参数说明

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| input\_weight | 输入 | 权重Tensor，为float类型。 |
| input\_grad | 输入 | 权重梯度Tensor，为float类型。 |
| weight\_decay | 输入 | 标量Tensor，为float类型。 |
| learning\_rate | 输入 | 标量Tensor，为float类型，表示学习率。 |
| hyperpara | 输入 | 标量，算子的超参，为float类型，一般设定为0.001。 |
| epsilon | 输入 | 标量，一般为很小的正数，防止分母为0，一般设定为1e\-5。 |
| use\_clip | 输入 | bool类型，默认为False。 当配置为True时，表示缩放系数需要限定在一定的范围内。 |
| name | 输入 | 网络层的名称。 |


#### 返回值

tensor：对输入的梯度进行更新后的输出梯度Tensor。


#### 调用示例

```
from npu_bridge.npu_init import *
layers = npu_ops.LARSV2(input_weight , input_grad, weight_decay, learning_rate)
```