【人工智能概論】 PyTorch中的topk、expand_as、eq方法
一. topk方法
1.1 簡介
- 對PyTorch中的tensor類型的數(shù)據(jù)都存在topk方法，其功能是按照要求取前k個最大值。
- 其最常用的場合就是求一個樣本被網(wǎng)絡(luò)認(rèn)為的前k種可能的類別。
- 舉例：
data = torch.tensor(
[[1,2,3],
[1,5,4],
[9,7,5]]
)
print(data)
data_topk , index_topk = data.topk(2,1,True,True)
print(data_topk)
print(index_topk)
1.2 參數(shù)詳解
torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) 文章來源：http://www.zghlxwxcb.cn/news/detail-499118.html
- 其中：
- input： 是待處理的tensor數(shù)據(jù)；
- k： 指明要前k個數(shù)據(jù)及其index；
- dim： 指定在什么維度上排序，默認(rèn)最后一維；
- largest： 如果為True，按照降序排序，如果為False，按照升序排序；
- sorted： 返回的結(jié)果是否按照順序排列；
- out： 可缺省，不要。
二. expand_as方法
- 用來擴(kuò)展tensor中某維數(shù)據(jù)的尺寸，將輸入tensor的維度擴(kuò)展為與指定tensor相同的尺寸。
- 舉例：
import torch
a = torch.tensor([[2], [3], [4]])
print(a)
b = torch.tensor([[2, 2], [3, 3], [5, 5]])
print(b.size())
c = a.expand_as(b)
print(c)
print(c.size())
文章來源地址http://www.zghlxwxcb.cn/news/detail-499118.html
三. eq方法
- torch.eq(tensor1, tensor2, out=None) || tensor1.eq(tensor2,out=None)：tensor1對應(yīng)的元素等于tensor2的元素會返回True，否則返回False。參數(shù)out表示為一個數(shù)或者是與第一個參數(shù)相同形狀和類型的tensor。
- 舉例：
Matrix_A = torch.tensor([1,2,3,4,5,3,1])
Matrix_B = torch.tensor([2,1,3,1,5,2,0])
print(Matrix_A.eq(Matrix_B))
到了這里，關(guān)于【人工智能概論】 PyTorch中的topk、expand_as、eq方法的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
