pytroch 函数学习

1.  tensor=torch.from_numpy(ndarray).

将ndarray转化为tensor

2 .     ndarray =tensor.numpy()

将tensor转化为ndarray

3.    x = torch.zeros(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)  

>>> torch.zeros(2, 3)
tensor([[ 0.,  0.,  0.],
        [ 0.,  0.,  0.]])

>>> torch.zeros(5)
tensor([ 0.,  0.,  0.,  0.,  0.]

初始化一个全零数组。

4.  y=torch.zeros_like(x)

构造一个形状和类型和x一致的全零数组

5. torch.ones(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

 torch.ones_like(input, dtype=None, layout=None, device=None, requires_grad=False) → Tensor

和上面的例子一样，只是这次构造的是全一数组

6. torch.arange(start=0, end, step=1, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

torch.range(start=0, end, step=1, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

构造一个顺序数组

>>> torch.arange(5)
tensor([ 0,  1,  2,  3,  4])
>>> torch.arange(1, 4)
tensor([ 1,  2,  3])
>>> torch.arange(1, 2.5, 0.5)
tensor([ 1.0000,  1.5000,  2.0000])

arange()和range()的区别是range()结果多了一位,并且range()需要起始点。

>>>a=torch.range(0,5)
>>>print a
tensor([ 0.,  1.,  2.,  3.,  4.,  5.])

7. torch.masked_select(input, mask, o