2021年9月17日
one_hot编码
摘要： src0 = torch.tensor([[ 0.], [1.], [2.], [3.], [4.], [5.], [1.], [2.], [3.], [3.], [0.], [1.], [4.]]) src = np.array(src0).squeeze() torch.eye(6)[src,: 阅读全文
posted @ 2021-09-17 01:14 呦呦南山 阅读(23) 评论(0) 推荐(0) 编辑