深度学习
王太乙
专业:计算机视觉
就业方向:计算机视觉/SLAM算法
年级:硕士研究生
展开
-
切片[:, None, None]的含义
[:, None, None]是改变数组维度的方式假设:a = np.array([1,2,3])进行下面操作a[:,None]结果为:Out[35]: array([[1], [2], [3]])也就是一维变成二维的了,如果:a[:,None,None]大家猜猜结果是什么?第一个冒号代表切片,把一维a的元素全部切完,然后第二个为None,表示把一维变成二维并且第二个维度上的值为0,第三个元素为None,表示增加一个维度,并且该维度值为0所以a[原创 2021-09-02 19:14:03 · 8168 阅读 · 1 评论 -
torch模块中NLLLoss损失与LogSoftmax激活层的关系
import torchimport cv2 as cvm = torch.nn.LogSoftmax(dim=1)loss = torch.nn.NLLLoss()Input = torch.randn(2, 2, requires_grad=True)print(Input)target = torch.tensor([0, 1])print("m(input)", m(Input))output = loss(m(Input), target)print("Target", tar原创 2021-08-26 14:37:10 · 763 阅读 · 0 评论