摘要: Tensor advanced operation ▪ Where ▪ Gather # where 返回的最终的tensor中的每一个值有可能来着A,也有可能来自B。 >torch.where(condition,A,B)->tensor 满足condition条件则该位置为A[..],否则为B[ 阅读全文
posted @ 2023-07-31 23:14 lipu123 阅读(11) 评论(0) 推荐(0) 编辑
摘要: statistics ▪ norm (范数) ▪ mean,sum (平均值,求和) ▪ prod (累乘) ▪ max, min, argmin, argmax ▪ kthvalue, topk(第k大) # norm(范式) 这里面有一范式和二范式。 一范式: $$||x||_1=\sum_k| 阅读全文
posted @ 2023-07-31 21:47 lipu123 阅读(159) 评论(0) 推荐(0) 编辑
摘要: Math operation ▪ Add/minus/multiply/divide(加减乘除) ▪ Matmul(矩阵相乘) ▪ Pow(次方) ▪ Sqrt/rsqrt(次方根) ▪ Round() add/minus/multiply/divide 这个使用的时候可以直接使用运算符"+,-,* 阅读全文
posted @ 2023-07-31 14:50 lipu123 阅读(53) 评论(0) 推荐(0) 编辑
摘要: Merge or split ▪ Cat(合并) ▪ Stack(合并) ▪ Split(拆分) ▪ Chunk(拆分) # 合并 ## cat 这个就是合并两个tensor 比如说有两个班级的成绩单,一个是1-4班的,一个是5-9班的,我们现在需要合并这两份成绩单。 ▪ Statistics ab 阅读全文
posted @ 2023-07-31 10:10 lipu123 阅读(73) 评论(0) 推荐(0) 编辑