_utils#
用于PyTorch Forecasting的辅助函数
函数
|
将函数应用于对象列表,或直接应用于传递的值(如果该值不是列表)。 |
|
计算在维度 |
|
连接 RNN 序列。 |
|
创建形状为 len(lenghts) x size 的布尔掩码。 |
|
分离对象 |
|
从经验上确定好的嵌入尺寸(公式来自fastai)。 |
|
对torch张量进行分组应用 |
|
创建预定义范围内整数的直方图 |
|
对掩码张量进行计算操作。 |
|
将对象移动到设备。 |
|
返回下一个最大小于或等于 |
|
沿着第一维堆叠张量,并在最后一维中填充以确保它们的大小相等。 |
|
使用 |
|
打印类名和参数。 |
|
将值或列表转换为值的列表。 |
|
解压RNN序列。 |
|
对张量的最后维度进行扩展,以匹配另一个张量的维度数量。 |
类
MixIn 给予 namedtuple 一些字典的访问能力 |
|
MixIn 为输出提供类似命名元组的访问能力,通过 |