pytorch每日一学19(torch.ones()与torch.ones_like())创建全为1的tensor

第19个方法

其实接下来这两个方法和我前面的torch.zeros()和torch.zeros_like()是完全相似的，只是一个创建全1，一个创建全0，所以大家可以仿照前面即可。

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

torch.ones_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format)->Tensor
1

例子如下：

>>> torch.ones(2, 3)
tensor([[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]])
>>> torch.ones(5)
tensor([ 1.,  1.,  1.,  1.,  1.])




>>> input = torch.empty(2, 3)
>>> torch.ones_like(input)
tensor([[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]])
1
2
3
4
5
6
7
8
9
10
11
12
13

• 详细介绍参照前面两个方法，参数也一致，链接如下torch.zeros()、以及torch.zeros_like()