Pytroch基本用法大全

1. 场景

        在改进模型过程中，会遇到一些基本的python、pytorch用法，做下记录

2. 用法

1. 如何查看列表的维度

import numpy as np
list = [1,2,3]
print(np.array(list).shape) # 查看列表维度

2. Tensor与Numpy相互转换

# Tensor转换为Numpy
a = torch.ones(5)
b = a.numpy()

# Numpy转换为Tensor
a = np.ones(5)
b = torch.from_numpy(a)

3. 统计numpy中某个元素的个数

# data最初是Numpy类型
print(np.sum(data == True))

4. Concat和add的区别

一般情况下，feature maps的结合有两种方法，一种是元素对应相加，简称add，另一种就是把特征图堆到一起来，简称concatenate。

Concat：张量拼接，会扩充两个张量的维度，增加通道数，
add：张量相加，张量直接相加，不会扩充维度。