代码拉取完成,页面将自动刷新
name | about | labels |
---|---|---|
Bug Report | Conv3d的入参dilation的depth不能被设置为除了1以外的数 | kind/bug |
开发vistr模型时,需要使用dilation为(2,2,2)的Conv3d空洞卷积算子。但是由于Conv3d算子的dilation的第一个数只能是1,即(1, 2, 2),所以无法实现。
Ascend
/GPU
/CPU
) / 硬件环境:Please delete the backend not involved / 请删除不涉及的后端:
/device GPU
Software Environment / 软件环境 (Mandatory / 必填):
-- MindSpore version (e.g., 1.7.0.Bxxx) : 1.7.0
-- Python version (e.g., Python 3.7.5) : 3.7.5
-- OS platform and distribution (e.g., Linux Ubuntu 16.04): euleros2u3
-- GCC/Compiler version (if compiled from source): 7.3.0
Excute Mode / 执行模式 (Mandatory / 必填)(PyNative
/Graph
):
Please delete the mode not involved / 请删除不涉及的模式:
/mode pynative
/mode graph
import mindspore.nn as nn
conv3d = nn.Conv3d(in_channels=3, out_channels=32, kernel_size=(4, 3, 3), dilation=(2,2,2))
Conv3d可以设置为dilation为(2,2,2)的空洞卷积。
登录 后才可以发表评论