SquaredDifference#

SquaredDifference operation performs element-wise subtraction operation with two given tensors applying multi-directional broadcast rules, after that each result of the subtraction is squared.

Before performing arithmetic operation, \(src_1\) and \(src_2\) are broadcasted if their shapes are different and auto_broadcast attributes is not none. Broadcasting is performed according to auto_broadcast value. After broadcasting SquaredDifference does the following with the input tensors:

\[\dst_i = (\src\_1_i - \src\_2_i)^2\]

Operation Attributes#

Attribute

Name

Description

Value Type

Supported

Values

Required or

Optional

auto_broadcast

Specifies rules used for auto-broadcasting of src tensors

string

none, numpy, (default)

Optional

Execution Arguments#

The inputs and outputs must be provided according to the below index order when constructing an operation.

Inputs#

Index

Argument Name

Required or Optional

0

src_1

Required

1

src_2

Required

Outputs#

Index

Argument Name

Required or Optional

0

dst

Required

Supported Data Types#

SquaredDifference operation supports the following data type combinations.

Src_1

Src_2

Dst

f32

f32

f32

bf16

bf16

bf16

f16

f16

f16