我是想同时使用了tensorflow的运算和pytorch的向量生成。因为pytorch生成的向量是有梯度的,所以必须用 .detach()来消除掉影响即可
比如:
而uembedding是pytorch生成的向量
uembedding.detach()
我是想同时使用了tensorflow的运算和pytorch的向量生成。因为pytorch生成的向量是有梯度的,所以必须用 .detach()来消除掉影响即可
比如:
而uembedding是pytorch生成的向量
uembedding.detach()