python
baekii
这个作者很懒,什么都没留下…
展开
-
【PyTorch基础】乘法torch.mul(),torch.nn()
torch.mul(a, b)是矩阵a和b对应位相乘,a和b的维度必须相等,相乘后返回结果维度也相同 torch.mm(a, b)是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b是(2, 3),返回的就是(1, 3)的矩阵 举个栗子: import torch a = torch.rand(1, 4) b = torch.rand(4, 3) c = torch.rand(1, 3) print(a,'\n',b,'\n',c) ab = torch.mm(a, b) abc = torch.m.原创 2020-09-20 14:57:56 · 908 阅读 · 0 评论 -
【PyTorch基础】torch.sum()
torch.sum()对输入的tensor数据的某一维度求和,一共两种用法 1.torch.sum(input, dtype=None) 2.torch.sum(input, list: dim, bool: keepdim=False, dtype=None) → Tensor input:输入一个tensor dim:要求和的维度,可以是一个列表 keepdim:求和之后这个dim的元素个数为1,所以要被去掉,如果要保留这个维度,则应当keepdim=True 举例说明: import torc原创 2020-09-20 14:47:43 · 2801 阅读 · 0 评论 -
Python函数装饰器详解
1. 装饰器 增强函数或类的功能的一个函数,可以装饰函数,也可以装饰类 1.1 举个栗子 def time_decorator(func): #装饰器 def wrapper(*args, **kwargs): #wrapper在第二节中具体讲 # 在函数前打印开始时间 start_time = time.time() print("start time is:", start_time) # 执行待装饰函数本身原创 2020-09-12 20:26:16 · 215 阅读 · 0 评论