gpt4 book ai didi

python - CNTK 卷积1d

转载 作者:行者123 更新时间:2023-11-28 17:14:35 24 4
gpt4 key购买 nike

我正尝试在 CNTK 中创建一个简单的卷积模型,如下所示

def create_model(hidden_dim, output_dim):
nn=C.layers.Sequential([ C.layers.Embedding(shape=50,name='embedding'),
C.layers.Convolution1D((40,),num_filters=5, activation=C.ops.relu),
C.layers.GlobalMaxPooling(),
C.layers.Dense(shape=40, activation=C.ops.tanh, init_bias=0.1),
C.layers.Dense(shape=2, activation=None, init_bias=0.1)
])
return nn

但我不断收到以下错误ValueError: 卷积图张量的秩必须为 1 或与输入张量相同。

最佳答案

我能够通过将 reduction_rank=0 作为参数添加到 Convolution1d 层来解决这个问题。

def create_model(hidden_dim, output_dim):
nn=C.layers.Sequential([ C.layers.Embedding(shape=50,name='embedding', **reduction_rank=0**),
C.layers.Convolution1D((40,),num_filters=5, activation=C.ops.relu),
C.layers.GlobalMaxPooling(),
C.layers.Dense(shape=40, activation=C.ops.tanh, init_bias=0.1),
C.layers.Dense(shape=2, activation=None, init_bias=0.1)
])
return nn

引自CNTK Layers Documentation

reduction_rank (int, defaults to 1) – 如果输入项是标量(输入没有深度轴),则设置为 0,例如以张量形状 (H,W) 而不是 (1,H,W) 存储的音频信号或黑白图像

我期待CNTK能够自动推断出这个东西

关于python - CNTK 卷积1d,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45069504/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com