摘要: 有时候遇到不支持maxpool3d的硬件或算子时候,可将其改成maxpool2d加上maxpool1d组合方式表示,经验证与maxpool3d结果完全一致,其实现细节如下: 代码: import torch class MaxPool3d_modify(torch.nn.Module): def _ 阅读全文
posted @ 2023-03-02 11:23 tangjunjun 阅读(188) 评论(0) 推荐(0) 编辑
https://rpc.cnblogs.com/metaweblog/tangjunjun