测试实用工具

xfail_if_not_implemented(msg='Not implemented', *, match=None)

xfail_if_not_found(msg='Not implemented')

requires_backend(*backends, reason=None)

excludes_backend(*backends, reason=None)

class ActualExpected(actual, expected)

基类: LazyComparison

用于测试断言的延迟字符串格式化程序。

id_from_inputs(inputs)

is_array(x)

assert_close(actual, expected, atol=1e-06, rtol=1e-06)

check_funsor(x, inputs, output, data=None)

检查维度和形状（忽略重新排序）。

xfail_param(*args, **kwargs)

make_einsum_example(equation, fill=None, sizes=(2, 3))

assert_equiv(x, y)

检查两个 funsors 在输入排列上是否等价。

rand(*args)

randint(low, high, size)

randn(*args)

random_scale_tril(*args)

zeros(*args)

ones(*args)

empty(*args)

random_tensor(inputs, output=Real)

创建一个具有给定输入和输出的随机 funsor.tensor.Tensor。

random_gaussian(inputs)

创建一个具有给定输入的随机 funsor.gaussian.Gaussian。

random_mvn(batch_shape, dim, diag=False)

生成一个具有给定形状的随机 torch.distributions.MultivariateNormal。

make_plated_hmm_einsum(num_steps, num_obs_plates=1, num_hidden_plates=0)

make_chain_einsum(num_steps)

make_hmm_einsum(num_steps)

iter_subsets(iterable, *, min_size=None, max_size=None)

class DesugarGetitem

基类: object

用于解语法糖 .__getitem__() 语法的辅助工具。

示例

>>> desugar_getitem[1:3, ..., None]
(slice(1, 3), Ellipsis, None)