set_output

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 tensor of the output with a specified index.

Prototype

1
set_output(self, index, output: Union[FlowMsg, np.ndarray, fw.FlowMsg], balance_config: BalanceConfig = None) -> int

Parameters

Parameter

Input/Output

Description

index

Input

Index of the specified output.

output

Input

Message of the specified output.

balance_config

Input

(Optional) Configurations related to balanced distribution.

Returns

Exception Handling

None

Constraints

None