- 博客(7)
- 收藏
- 关注
原创 ‘tensorrt.tensorrt.Builder‘ object has no attribute ‘build_cuda_engine‘
问题简述:AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'原因:tensorrt版本大于等于7时,不再使用build_cuda_engine。解决办法:print('[info] Building an engine...')if TRT_VERSION_MAJOR >= 7: # Create an optimization profile (se
2022-01-13 20:04:04 8314 7
原创 PyTorch中scatter()和scatter_()函数的作用
本文讲的是我对PyTorch中scatter()函数的理解。原创,转载请标明来源。一言以蔽之:修改tensor中的指定位置的值。函数scatter(dim, index, src)dim: 索引的维度。按照i, j, k, ...的哪个方向进行索引 index: 索引。可以是一个tensor,存储需要改的元素的位置的tensor src: 用src中的值来修改。可以是tensor;可以是一个数字,用同样的数字写入tensorscatter() 和 scatter_() 函数..
2020-10-04 15:20:44 1517
原创 Python | **,**kwargs,*,*args的作用
(Python版本3.7.4)1. **与**kwargs1.1 **的作用**的作用是将字典中的内容作为参数传入函数中。# 举例def print1(a, b, c): print(a, b, c)# 定义一个字典dict = {'a':1, 'b':2, 'c':3}# 传入函数中print1(**dict)函数返回:1 2 3在print1(...
2020-04-08 10:43:09 157
原创 Ubuntu没有声音的解决历程分享
配置信息:win10 + Ubuntu16.04LTS双系统。华硕笔记本电脑。有自带的扬声器和耳机孔。win10的声音正常,Ubuntu没有声音。不同的软硬件的情况不同,本文的解决方法并非在你的电脑上100%可用。解决历程:1. 根据网上的教程,尝试了安装 pavucontrol :sudo apt install pavucontrol之后打开pavucontro...
2020-03-21 01:44:28 3120 1
原创 torch.gather()的理解
函数定义torch.gather(input, dim, index, out=None)其中,input(tensor): 待操作数,源张量dim(int): 维度。index(LongTensor): 索引下标out: 输出的张量举例dim=0输入:a = torch.arange(0,16).view(4,4)print(a)index...
2020-03-18 03:08:17 499 1
原创 jupyter notebook 中方括号(中括号)在markdown中不能正常显示的处理
在jupyternotebook中,若在markdown模式下输入a[0][0] = 0往往不会显示正确的结果。这时可以使用 转义 (反斜杠)来解决这个问题:a\[0\]\[0\] = 0就可以正确显示了。另外,因为jupyter notebook是一个开源的Web应用程序,所以使用html转义字符也可以解决:( [ ] 的HTML转义字符分别为[和...
2020-03-18 02:32:59 3196
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人