哪有什么岁月静好,不过是有人替你负重前行!
摘要: 1.torch.where(condition,x,y) 有两种情况: 1.包含三个参数,如果condition为true,则 value值为x; 如果condition为false,则 value值为y; 2.若只包含condition参数,若进行判断的是一个n维张量,则返回一个元组,包括n个te 阅读全文
posted @ 2022-05-03 15:14 Lhiker 阅读(252) 评论(0) 推荐(0) 编辑