比较与torch.zeros的差异

torch.zeros

torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor

更多内容详见torch.zeros。

mindspore.ops.zeros

mindspore.ops.zeros(size, dtype=dtype) -> Tensor

更多内容详见mindspore.ops.zeros。

差异对比

PyTorch：生成大小为 *size 的填充值为0的Tensor。

MindSpore：生成shape为 size 的填充值为0的Tensor。

分类	子类	PyTorch	MindSpore	差异
参数	参数1	size	size	MindSpore支持int、tuple或者Tensor类型的输入
	参数2	out	-	不涉及
	参数3	dtype	dtype	无差异
	参数4	layout	-	不涉及
	参数5	device	-	不涉及
	参数6	requires_grad	-	不涉及

代码示例1

两API实现功能一致，用法相同。

# PyTorch
import torch
from torch import tensor

output = torch.zeros(2, 2, dtype=torch.float32)
print(output.numpy())
# [[0. 0.]
#  [0. 0.]]

# MindSpore
import numpy as np
import mindspore
import mindspore.ops as ops
import mindspore as ms
from mindspore import Tensor

output = ops.zeros((2, 2), dtype=ms.float32).asnumpy()
print(output)
# [[0. 0.]
#  [0. 0.]]