set_output

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

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