SetBalanceWeight

Applicable Products

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	x
Atlas inference products	x
Atlas training products	x

Function Description

Sets the weight of balanced distribution.

Prototype

void SetBalanceWeight(const BalanceWeight &balanceWeight) = 0

Command-Line Options

Parameter

Input/Output

Description

balanceWeight

Input

Weight of balanced distribution

struct BalanceWeight {
int32_t rowNum = 0;  // Number of rows in the weight matrix
int32_t colNum = 0;   // Number of columns in the weight matrix
const int32_t *matrix = nullptr; // Weight matrix, which contains arrays of rowNum rows and colNum columns. If this value is null, the values in the matrix are all 1.
};

Returns

None

Exception Handling

None

Constraints

None

Parent topic: Class BalanceConfig