vexp

Description

Performs the natural exponential operation on a tensor element-wise and returns e^x.

Prototype

vexp(raw_tensor)

Parameters

raw_tensor: a tvm.tensor for the input tensor

Atlas training product: supports float16 and float32.

Atlas inference product: supports float16 and float32.

Atlas 200I/500 A2 inference product: supports float16 and float32.

Atlas A2 training product/Atlas A2 inference product: supports float16 and float32.

Atlas A3 training product/Atlas A3 inference product: supports float16 and float32.

Returns

res_tensor: a tvm.tensor for the result tensor

Restrictions

None

Applicability

Atlas training product

Atlas inference product

Atlas 200I/500 A2 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas A3 training product/Atlas A3 inference 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.vexp(data)