pytorch
candy134834
这个作者很懒,什么都没留下…
展开
-
pytorch mask_filled用法
#将 mask必须是一个 ByteTensor 而且shape必须和 a一样 并且元素只能是 0或者1 ,是将 mask中为1的 元素所在的索引,在a中相同的的索引处替换为 value ,mask value必须同为tensor a=torch.tensor([1,0,2,3]) # a.masked_fill(mask = torch.ByteTensor([1,1,0,0]),...原创 2018-11-28 23:43:46 · 31549 阅读 · 3 评论 -
torch.narrow()
import torch.tensor x=torch.rand(5,6)x tensor([[0.4606, 0.0850, 0.8009, 0.3972, 0.9548, 0.5982], [0.4821, 0.9446, 0.5145, 0.8125, 0.3122, 0.9756], [0.8747, 0.7186, 0.3945, 0.4090, 0...原创 2019-01-23 07:56:20 · 1448 阅读 · 0 评论