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