原文:torch常用的函数

torch.cat :是将两个张量 tensor 拼接在一起。 tensor.expand as :把一个tensor变成和函数括号内一样形状的tensor tensor.narrow dim,index,number :dim 取行 列 index 从索引index开始取 number 取的行数 列数 contiguous :在调用view 之前需要先调用contiguous tensor.c ...

2020-07-21 11:05 0 1071 推荐指数:

查看详情

PyTorch常用函数torch.ge; torch.gt; torch.le; equal; eq

torch.ge torch.ge(input, other, out=None) → Tensor 逐元素比较input和other,即是否 input>=otherinput>=other。 如果两个张量有相同的形状和元素值,则返回True ,否则 False。 第二个参数可以为 ...

Sun Feb 23 03:24:00 CST 2020 0 3392
torch 中的损失函数

NLLLoss 和 CrossEntropyLoss 在图片单标签分类时,输入m张图片,输出一个m*N的Tensor,其中N是分类个数。比如输入3张图片,分3类,最后的输出是一个3*3的Tens ...

Tue Jul 21 01:43:00 CST 2020 0 687
torch.chunk()函数

将tensor按dim方向分割成chunks个tensor块,返回的是一个元组。 示例: ...

Wed Dec 02 06:41:00 CST 2020 0 1024
torch.clamp() 函数

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

Sat Sep 05 20:19:00 CST 2020 0 2406
torch.max()函数

一、_, predicted = torch.max(outputs.data, 1) 那么,这里的 下划线_ 表示什么意思? 首先,torch.max()这个函数返回的是两个值,第一个值是具体的value(我们用下划线_表示),第二个值是value所在的index(也就是predicted ...

Wed Mar 09 18:34:00 CST 2022 0 932
torch topk函数

这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 比如,三行两列,3个样本,2个类别。 ...

Thu Mar 24 18:37:00 CST 2022 0 2130
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM