ops.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:speech 作者: igul222 项目源码 文件源码
def Embedding(name, n_symbols, output_dim, indices):
    vectors = lib.param(
        name,
        numpy.random.randn(
            n_symbols, 
            output_dim
        ).astype(theano.config.floatX)
    )

    output_shape = [
        indices.shape[i]
        for i in xrange(indices.ndim)
    ] + [output_dim]

    return vectors[indices.flatten()].reshape(output_shape)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号