pytorch-tensor高级OP

Tensor advanced operation ? Where ? Gather # where 返回的最终的tensor中的每一个值有可能来着A,也有可能来自B。 >torch.where(condition,A,B)->tensor 满足condition条件则该位置为A[..],否则为B[
posted @ 2023-07-31 23:14  哎呦哎(iui)  阅读(6)  评论(0)  编辑  收藏  举报