
torchcubicspline:自然三次样条内插器,具备批处理、GPU支持、缺失值处理及导数评估与反向传播功能
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
TorchCubicSpline是专为PyTorch设计的高效自然三次样条插值库,支持批量操作、GPU加速、缺失数据处理和求导与反向传播。
火炬花键使用PyTorch插值自然三次样条。 包括以下功能: 批处理通过PyTorch支持GPU和反向传播; 支持缺失值(表示为NaN)。 安装方法如下:
```
pip install git+https://github.com/patrick-kidger/torchcubicspline.git
```
简单示例代码:
```python
import torch
from torchcubicspline import natural_cubic_spline_coeffs, NaturalCubicSpline
length = 7
channels = 3
t = torch.linspace(0, 1, length)
x = torch.rand(length, channels)
```
全部评论 (0)
还没有任何评论哟~


