_utils#

用于PyTorch Forecasting的辅助函数

函数

apply_to_list(obj, func)

将函数应用于对象列表,或直接应用于传递的值(如果该值不是列表)。

autocorrelation(输入[, 维度])

计算在维度 dim 的样本的自相关。

concat_sequences(序列)

连接 RNN 序列。

create_mask(size, lengths[, inverse])

创建形状为 len(lenghts) x size 的布尔掩码。

detach(x)

分离对象

get_embedding_size(n[, max_size])

从经验上确定好的嵌入尺寸(公式来自fastai)。

groupby_apply(keys, values[, bins, ...])

对torch张量进行分组应用

integer_histogram(数据[, 最小值, 最大值])

创建预定义范围内整数的直方图

masked_op(张量[, 操作, 维度, 掩码])

对掩码张量进行计算操作。

move_to_device(x, device)

将对象移动到设备。

next_fast_len(大小)

返回下一个最大小于或等于 n >= size 的数字,其质因数全部为 2、3 或 5。

padded_stack(张量[, 边, 模式, 值])

沿着第一维堆叠张量,并在最后一维中填充以确保它们的大小相等。

profile(函数, profile_fname[, filter, ...])

使用 vmprof 对给定函数进行性能分析。

repr_class(obj, attributes[, ...])

打印类名和参数。

to_list(value)

将值或列表转换为值的列表。

unpack_sequence(序列)

解压RNN序列。

unsqueeze_like(张量, 像)

对张量的最后维度进行扩展,以匹配另一个张量的维度数量。

InitialParameterRepresenterMixIn()

OutputMixIn()

MixIn 给予 namedtuple 一些字典的访问能力

TupleOutputMixIn()

MixIn 为输出提供类似命名元组的访问能力,通过 to_network_output() function