assert_allclose

比较指定tensor的值与期望值是否在一定精度范围内。

def assert_allclose(tensor, desired, tol=[1e-7, 1e-7]):

参数名	类型	说明
tensor	Tensor	The tensor to be compared. 需要进行比对的Tensor。
desired	numpy.ndarray	期望数据。
tol	List[float, float]	tol=[rtol, atol] 即：相对误差（rtol）与绝对误差（atol）组成的列表。默认值为：[1e-7, 1e-7]

参数名

类型

说明

tensor

Tensor

The tensor to be compared.

需要进行比对的Tensor。

desired

numpy.ndarray

期望数据。

tol

List[float, float]

tol=[rtol, atol]

即：相对误差（rtol）与绝对误差（atol）组成的列表。

默认值为：[1e-7, 1e-7]

无。

assert_allclose接口仅支持需要比对的数据类型为Tensor，期望数据类型为numpy.ndarray的场景。

若您需要比对的数据类型为numpy.ndarray，可调用tvm.testing.assert_allclose接口。

父主题： TBE DSL Testing API