PyTorch教程之Tensors
>>> import torch
>>> x = torch.Tensor(2,3)
>>> x
tensor([[0., 0., 0.],
[0., 0., 0.]])
>>> y=torch.rand(3,4)
>>> y
tensor([[0.4464, 0.8380, 0.2633, 0.6285],
[0.9167, 0.4549, 0.4253, 0.6597],
[0.0015, 0.4274, 0.2199, 0.3969]])
>>> y.size()
torch.Size([3, 4])
>>> a = y.size()[0]
>>> a
3
>>> b = y.size()[1]
>>> b
4
>>> z = torch.rand(3,4)
>>> z
tensor([[0.3165, 0.9564, 0.0270, 0.6598],
[0.6044, 0.8573, 0.3340, 0.8657],
[0.0120, 0.6371, 0.1377, 0.8685]])
>>> y+z
tensor([[0.7629, 1.7944, 0.2903, 1.2883],
[1.5211, 1.3122, 0.7593, 1.5254],
[0.0136, 1.0645, 0.3576, 1.2654]])
>>> torch.add(y,z)
tensor([[0.7629, 1.7944, 0.2903, 1.2883],
[1.5211, 1.3122, 0.7593, 1.5254],
[0.0136, 1.0645, 0.3576, 1.2654]])
>>> y.add_(z)
tensor([[0.7629, 1.7944, 0.2903, 1.2883],
[1.5211, 1.3122, 0.7593, 1.5254],
[0.0136, 1.0645, 0.3576, 1.2654]])
>>> y
tensor([[0.7629, 1.7944, 0.2903, 1.2883],
[1.5211, 1.3122, 0.7593, 1.5254],
[0.0136, 1.0645, 0.3576, 1.2654]])
>>> y[:,1]
tensor([1.7944, 1.3122, 1.0645])
>>> y[1,:]
tensor([1.5211, 1.3122, 0.7593, 1.5254])
>>>