PyTorch 0.4.1 发布

PyTorch入门实战教程

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
  • 改进了文档的阅读体验

完整版的发布日志请查看这里

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-0-4-1-release/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部