0
点赞
收藏
分享

微信扫一扫

python的小tips

西曲风 2022-04-14 阅读 45
python

python的小tips

1 利用pytorch生成掩膜,去掉不想要的值

import torch
import numpy as np
a = [1,0,1,0,1,0]
b = [1,2,3,4,5,6]
a = torch.from_numpy(np.array(a))
b = torch.from_numpy(np.array(b))
c = b[a>0]
举报

相关推荐

0 条评论