目錄
- 1.normal
- 2. mean, sum, min, max, prod
- 3.argmin, argmax
- 4. topk kthvalue
- 5. compare
1.normal
torch.normal(mean, std, *, generator=None, out=None) → Tensor
返回一個張量,其中的每個元素隨機來自獨立的標準正態分布。這些分布具有給定的均值和標準差。
參數:
mean (Tensor) – 每個元素tensor的均值
std (Tensor) –每個元素tensor的標準差
generator (torch.Generator, optional)-用于采樣的偽隨機數生成器
輸出:
結果tensor
>>> torch.normal(mean=torch.arange(1., 11.), std=torch.arange(1, 0, -0.1))
tensor([ 1.0425, 3.5672, 2.7969, 4.2925, 4.7229, 6.2134,8.0505, 8.1408, 9.0563, 10.0566])
torch.normal(mean, std=1.0, *, out=None) → Tensor
類似于上面的函數,但是所有抽取到的元素共享相同的方差(標準差)
參數:
mean (Tensor) – 每個元素tensor的均值
std (Tensor) –每個元素tensor的標準差
輸出:
結果tensor
>>> torch.normal(mean=0.5, std=torch.arange(1., 6.))
tensor([-1.2793, -1.0732, -2.0687, 5.1177, -1.2303])
torch.normal(mean, std, size, *, out=None) → Tensor
類似于上面的函數,但是所有抽取到的元素共享相同的均值和方差(標準差)產生的tensor大小由size指定
參數:
size (int…) – 定義的一個輸出tensor shape的整形序列
輸出:
結果tensor
>>> torch.normal(2, 3, size=(1, 4))
tensor([[-1.3987, -1.9544, 3.6048, 0.7909]])
2. mean, sum, min, max, prod
torch.mean(input, *, dtype=None) → Tensor
返回輸出tensor所有元素的的均值
dtype-如果指定,操作執行前會將輸入張量轉換為dtype。這有助于防止數據類型溢出。默認值:None。
torch.prod(input, *, dtype=None) → Tensor
返回輸入張量中所有元素的乘積
torch.sum(input, *, dtype=None) → Tensor
返回輸入張量中所有元素的之和
torch.min(input) → Tensor
返回輸入張量中所有元素的最小值
torch.max(input) → Tensor
返回輸入張量中所有元素的最大值
3.argmin, argmax
torch.argmax(input) → LongTensor
torch.argmax(input, dim, keepdim=False) → LongTensor
返回輸入tensor中所有元素最大值的索引
torch.argmin(input) → LongTensor
torch.argmin(input, dim, keepdim=False) → LongTensor
返回輸入tensor中所有元素最小值的索引
參數:
input (Tensor) –輸入tensor
dim (int) –用于降低維度。如果設置為空(None),則返回展平后的輸入中的最大值索引位置(argmax)作為結果。
keepdim (bool) – 輸出張量是否會保留原始維度
dim keepdim
4. topk kthvalue
torch.topk(input, k, dim=None, largest=True, sorted=True, *, out=None)
返回輸出tensor指定維度上的k個最大值
參數:
input (Tensor) – 輸入張量
k (int) – k個最大值
dim (int, optional) – 指定維度
largest (bool, optional) – 控制返回最大值還是最小值
sorted (bool, optional) – 是否對返回結果排序
torch.kthvalue(input, k, dim=None, keepdim=False, *, out=None)
返回一個命名元組(values, indices) values 是輸入張量在指定維度dim中每一行元素的k個最小值,indices是每個返回元素的索引位置
5. compare
>, >=, <, <=, !=, ==等價于torch.gt torch.ge torch.lt torch.le torch.ne torch.eq
torch.equal(input, other) → bool
如果input和other具有相同的size和元素,那么返回Ture否則返回False