- 博客(2)
- 资源 (1)
- 收藏
- 关注
原创 pytorch中获取指定位置元素
这段代码的应用场景是:某个batch的sentence,有的经过了padding操作,如果获取每句话中实际的最后一个单词。A = torch.Tensor([[[2, 3, 1], [1, 4, 0], [1, 0, 0]], [[2, 2, 0], [2, 0, 0], [3, 1, 4]]])print(A.size())B = torch.Tensor([[3, 2, 1], [...
2018-10-31 21:28:44 19768
原创 pytorch中查看可训练参数
pytorch中我们有时候可能需要设定某些变量是参与训练的,这时候就需要查看哪些是可训练参数,以确定这些设置是成功的。 pytorch中model.parameters()函数定义如下: def parameters(self): r"""Returns an iterator over module parameters. This is typ...
2018-10-15 12:30:37 21909
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人