pytorch中的...

记录一个pytorch中的小把戏吧，最近又看了一遍yolo3代码，发现yolo3和ssd代码里都出现

        pred_boxes = FloatTensor(prediction[..., :4].shape)
        pred_boxes[..., 0] = x.data + self.grid_x
        pred_boxes[..., 1] = y.data + self.grid_y
        pred_boxes[..., 2] = torch.exp(w.data) * self.anchor_w
        pred_boxes[..., 3] = torch.exp(h.data) * self.anchor_h

表示之前看过也尝试查了，也在jupyter 里面试过了，然而我这个榆木脑子再看又忘了。记录一下吧，菜鸡的日子太难过了。有时间我也写一篇yolo的代码分析吧，立个flag，2020年写50片博客随机更新在Gray’s blog里面,有空的话合并。
实验：

import torch
b=torch.Tensor([[[[1,3],[11,33],[111,333]],[[2,4],[22,44],[222, 444]]]])
b

out:
tensor([[[  1.,   3.],
         [ 11.,  33.],
         [111., 333.]],

        [[  2.,   4.],
         [ 22.,  44.],
         [222., 444.]]])

b[..., 0]

out:
b[..., 0]
b[..., 0]
tensor([[[  1.,  11., 111.],
         [  2.,  22., 222.]]])

哈哈，就是最前面的那一列。虽然是小问题，也要多多记录啊