PyTorch 0.4 刚刚发布不久,官方又发布了 0.4.1。我们来看一下都有哪些改变吧。
重大更新
torch.stft
的参数列表改变了- 之前:
stft(signal, frame_length, hop, fft_size=None, normalized=False, onesided=True, window=None, pad_end=0)
- 现在:
stft(input, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True)
torch.stft
内部采用傅立叶变换(FFT),速度提升了很多
- 之前:
torch.slice
被移除torch.arange
默认参数类型浮点数是dtype
; 整数是int64
torch.nn.functional.embedding_bag
旧的参数 embedding_bag(weight, input, …) 已被弃用, 应该使用 embedding_bag(input, weight, …)torch.nn.functional.sigmoid
和torch.nn.functional.tanh
被弃用,应该是用torch.sigmoid
和torch.tanh
- 广播的一个小改变:
[1] x [0]
现在的结果是[0]
(以前是[1]
)
其他改进
- 新的功能
- 神经网络
- Adaptive Softmax, Spectral Norm, 等.
- 操作
- torch.bincount, torch.as_tensor, …
- torch.distributions
- Half Cauchy, Gamma Sampling, …
- 其它
- 自动检测不规则数 (比如 NaNs, 等.)
- 神经网络
- 性能
- CPU 的操作性能有所提升
- 修复了一些其它的 bug
- 改进了文档的阅读体验
完整版的发布日志请查看这里。
本站微信群、QQ群(三群号 726282629):