BalanceConfig Constructor

Applicability

Product

Supported

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

x

Atlas inference product

x

Atlas training product

x

Function Description

Constructs an object of class BalanceConfig.

Prototype

1
__init__(self, row_num: int, col_num: int, affinity_policy: AffinityPolicy = AffinityPolicy.NO_AFFINITY) -> None

Parameters

Parameter

Input/Output

Description

row_num

Input

Number of rows in the weight matrix.

col_num

Input

Number of columns in the weight matrix.

affinity_policy

Input

Affinity policy. The default value is non-affinity.

Returns

Object of class BalanceConfig

Exception Handling

None

Constraints

None