码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch的基础记录

时间:2019-02-24 23:07:08      阅读:349      评论:0      收藏:0      [点我收藏+]

标签:NPU   out   add   tensor   函数   包含   div   one   ddc   

torch.abs(input, out=None): 计算输入张量的每个元素的绝对值

torch.acos(input, out=None): 返回一个新张量,包含输入张量每个元素的反余弦函数

torch.add(input, value, out=None): 对输入张量input逐元素加上标量值value,并返回结果到一个新的张量out,即out=tensor+value

torch.addcdiv(tensor, value=1, tensor1, tensor2, out=None): 用tensor2对tensor1逐元素相除,然后乘以标量值value并加到tensor,张量的形状不需要匹配,但元素数量必须一致

torch.addcmul(tensor, value=1, tensor1, tensor2, out=None): 用tensor2对tensor1逐元素相乘,然后乘以标量值value并加到tensor,张量的形状不需要匹配,但元素数量必须一致

torch.asin(input, out=None): 返回一个新张量,包含输入张量每个元素的反正弦函数

torch.atan(input, out=None): 返回一个新张量,包含输入张量每个元素的反正切函数

torch.atan2(input1, input2, out=None): 返回一个新张量,包含两个输入张量的反正切函数

pytorch的基础记录

标签:NPU   out   add   tensor   函数   包含   div   one   ddc   

原文地址:https://www.cnblogs.com/czz0508/p/10428400.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!