numpy、pytorch记录

1、expand函数
2、expand_as函数

>>> import torch
>>> a = torch.tensor([[2],[3],[4]])
>>> a_size = a.size()
>>> b = torch.tensor([[0,0],[10,10],[100,100]])
>>> b_size = b.size()
>>> print(a,a_size, '    \n',b,b_size)
tensor([[2],
        [3],
        [4]]) torch.Size([3, 1])     
 tensor([[  0,   0],
        [ 10,  10],
        [100, 100]]) torch.Size([3, 2])
>>> a.expand(b_size)
tensor([[2, 2],
        [3, 3],
        [4, 4]])
>>> a.expand_as(b)
tensor([[2, 2],
        [3, 3],
        [4, 4]])

3、b = numpy.clip(a,min,max)
将a中的元素限制在min,max的闭区间中

4、np.concatenate((a,b,c),axis=0) # 默认情况下，axis=0可以不写
拼接 ： = array([a,b,c])

5、images = images.transpose(0, 3, 1, 2) #交换列，原来为 0，1，2，3 现在为0 3 1 2