In [ ]:
import torch
In [ ]:
a = torch.arange(10)
b = torch.arange(10)*10
print( a, b )
In [ ]:
a+b
In [ ]:
a.shape
In [ ]:
a[None].shape
In [ ]:
a[:,None].shape
In [ ]:
c = a[:,None]
c.shape
In [ ]:
c[:,0]
In [ ]:
d = a[None]
d.shape
In [ ]:
d[0]
In [ ]:
d[0,:]
In [ ]:
torch.ones([3,2]) + 10
In [ ]:
torch.ones([3,2]) + torch.ones([2,3])
In [ ]:
torch.ones([3,1]) + torch.ones([1,3])
In [ ]:
a[None,:] + b[:,None]
In [ ]:
a[None,:].repeat(10,1)
In [ ]:
b[:,None].repeat(1,10)
In [ ]:
a[None,:].repeat(10,1) + b[:,None].repeat(1,10)
In [ ]:
a[None,:] * b[:,None]
In [ ]: