335 Star 1.5K Fork 859

MindSpore / docs

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
bartlett_window.md 1.97 KB
一键复制 编辑 原始数据 按行查看 历史
luojianing 提交于 2023-07-21 15:16 . replace target=blank

比较与torch.bartlett_window的功能差异

查看源文件

torch.bartlett_window

torch.bartlett_window(
    window_length,
    periodic=True,
    *,
    dtype=None,
    layout=torch.strided,
    device=None,
    requires_grad=False
) -> Tensor

更多内容详见torch.bartlett_window

mindspore.ops.bartlett_window

mindspore.ops.bartlett_window(
    window_length,
    periodic=True,
    dtype=mstype.float32
) -> Tensor

更多内容详见mindspore.ops.bartlett_window

差异对比

PyTorch:返回size与window_length相同的巴特兰窗,periodic参数确定返回窗口是否会删除对称窗口的最后一个重复值。

MindSpore:MindSpore此API实现功能与PyTorch基本一致,精度稍有差异。

分类 子类 PyTorch MindSpore 差异
参数 参数1 window_length window_length PyTorch中为int类型,MindSpore中为Tensor
参数2 periodic periodic -
参数3 dtype dtype -
参数4 layout - 不涉及
参数5 device - 不涉及
参数6 requires_grad - MindSpore无此参数,默认支持反向求导

代码示例1

# PyTorch
import torch

torch_output = torch.bartlett_window(5, periodic=True)
print(torch_output.numpy())
#[0.         0.4        0.8        0.79999995 0.39999998]

# MindSpore
import mindspore
from mindspore import Tensor

window_length = Tensor(5, mindspore.int32)
ms_output = mindspore.ops.bartlett_window(window_length, periodic=True)
print(ms_output.asnumpy())
#[0.  0.4 0.8 0.8 0.4]
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r2.0

搜索帮助