2025年2月25日
pytorch将卷积展开成二维形式
摘要： import torch import torch.nn as nn def get_inp(inp,stride,kernel_size): channels = inp.shape[0] x_len = inp.shape[1] y_len = inp.shape[2] inp_list = [ 阅读全文
posted @ 2025-02-25 15:52 心比天高xzh 阅读(13) 评论(0) 推荐(0)