pytorch 计算topk_『PyTorch』第五弹_深入理解Tensor对象_中下：数学计算以及numpy比较_&_广播原理简介...

最新推荐文章于 2023-05-18 11:14:14 发布

weixin_39525865

最新推荐文章于 2023-05-18 11:14:14 发布

阅读量114

点赞数

文章标签： pytorch 计算topk

本文链接：https://blog.csdn.net/weixin_39525865/article/details/111510151

版权

本文介绍了PyTorch中的Tensor对象的数学运算，包括逐元素操作如 clamp、cos、取模和平方，归并操作如 sum 和 cumsum，以及比较操作如 topk 和 max。此外，还探讨了Tensor与Numpy的内存共享，以及广播原理的模拟和expand方法的应用。

摘要由CSDN通过智能技术生成

一、简单数学操作

1、逐元素操作

t.clamp(a,min=2,max=4)近似于tf.clip_by_value(A, min, max),修剪值域。

a = t.arange(0,6).view(2,3)

print("a:",a)

print("t.cos(a):",t.cos(a))

print("a % 3:",a % 3) # t.fmod(a, 3)

print("a ** 2:",a ** 2) # t.pow(a, 2)

print("t.clamp(a, min=2, max=4)",t.clamp(a,min=2,max=4))

0 1 2

3 4 5