theano set_subtensor

来源:互联网 发布:农村金融去哪找数据 编辑:程序博客网 时间:2024/06/05 11:33
emb_shifted = tensor.zeros_like(emb) #生成和 emb 形状相同的零矩阵
emb_shifted = tensor.set_subtensor(emb_shifted[1:], emb[:-1]) #set_subtensor 用于给tensor赋值
emb = emb_shifted #第一层全部变为 0
原创粉丝点击