2020年9月5日

pytorch中函数名后有下划线和没有下划线

摘要: 一般来说,在pytorch中如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型,所谓in-place类型是指在一个tensor上操作了之后,是直接修改了这个tensor,还是返回一个新的tensor,而旧的tensor并不修改。 例如: clamp和clamp_ x = 阅读全文

posted @ 2020-09-05 12:23 cltt 阅读(623) 评论(0) 推荐(0) 编辑

torch.clamp() 函数

摘要: torch.clamp(input, min, max, out=None) → Tensor 将输入input张量每个元素的夹紧到区间 [min,max][min,max],并返回结果到一个新张量。 操作定义如下: | min, if x_i < miny_i = | x_i, if min <= 阅读全文

posted @ 2020-09-05 12:19 cltt 阅读(5523) 评论(0) 推荐(0) 编辑

pytorch中tensor.mean(axis, keepdim)

摘要: 1 import numpy as np 2 import torch 3 4 x=[ 5 [[1,2,3,4], 6 [5,6,7,8], 7 [9,10,11,12]], 8 9 [[13,14,15,16], 10 [17,18,19,20], 11 [21,22,23,24]] 12 ] 1 阅读全文

posted @ 2020-09-05 10:55 cltt 阅读(2341) 评论(0) 推荐(0) 编辑

导航