卷积神经网络中的卷积操作
利用张量操作实现卷积# 利用张量操作实现卷积import torcha = torch.arange(16).view(4, 4)print(a)'''tensor([[ 0, 1, 2, 3], [ 4, 5, 6, 7], [ 8, 9, 10, 11], [12, 13, 14, 15]])'''b = a.unfold(0, 3, 1) # 按照行,以每三个元素,跨步为1进行展开print(b)'''tensor(
原创
2021-12-22 20:26:46 ·
836 阅读 ·
0 评论