vabs

Description

Performs the absolute value operation on a tensor element-wise and returns |x|.

Prototype

vabs(raw_tensor)

Parameters

raw_tensor: a tvm.tensor for the input tensor

Atlas 200/300/500 Inference Product: supports float16 and float32.

Atlas Training Series Product: supports float16 and float32.

Returns

res_tensor: a tvm.tensor for the result tensor

Restrictions

None

Applicability

Atlas 200/300/500 Inference Product

Atlas Training Series Product

Example

from tbe import tvm
from tbe import dsl
shape = (1024,1024)
input_dtype = "float16"
data = tvm.placeholder(shape, name="data", dtype=input_dtype)
res = dsl.vabs(data)