PyTorch 0.4.1 发布

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入门实战教程
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论