是的,您需要使用5维张量 Conv3d 。默认情况下,张量格式应该是 NCDHW 哪里:
NCDHW
'N' - 批量大小, 'C' - 频道, 'H' - 身高 'W' - 宽度 'D' - 深度。
卷积应用于'D','H'和'W'维度。
因此,如果您缺少通道维度(并且使用的是灰度数据),则可以创建此维度:
# a.shape is (64, 11, 11, 21) a = mx.nd.random.uniform(shape=(64, 11, 11, 21)) # adding 'channel' dimension a.expand_dims(1) # new a.shape is (64, 1, 11, 11, 21)