SetBalanceWeight

Applicability

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

1
void SetBalanceWeight(const BalanceWeight &balanceWeight) = 0

Command-Line Options

Parameter

Input/Output

Description

balanceWeight

Input

Weight of balanced distribution.

1
2
3
4
5
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

Troubleshooting

None

Constraints

None