PyTorch torch存儲(chǔ)

原文：PyTorch torch存儲(chǔ)

torch.Storage是單個(gè)數(shù)據(jù)類型的連續(xù)一維數(shù)組。

每個(gè) torch.Tensor 都有對(duì)應(yīng)的相同數(shù)據(jù)類型的存儲(chǔ)。

class torch.FloatStorage( * args , ** kwargs )

bfloat16()

將此存儲(chǔ)空間轉(zhuǎn)換為 bfloat16 類型

bool()

將此存儲(chǔ)轉(zhuǎn)換為布爾型

byte()

將此存儲(chǔ)空間轉(zhuǎn)換為字節(jié)類型

char()

將此存儲(chǔ)空間轉(zhuǎn)換為 char 類型

clone()

返回此存儲(chǔ)的副本

copy_()

cpu()

返回此存儲(chǔ)的 CPU 副本(如果尚未在 CPU 上）

cuda(device=None, non_blocking=False, **kwargs)

返回此對(duì)象在 CUDA 內(nèi)存中的副本。

如果此對(duì)象已經(jīng)在 CUDA 內(nèi)存中并且在正確的設(shè)備上，則不執(zhí)行任何復(fù)制，并返回原始對(duì)象。

參數(shù)

• 設(shè)備 (python：int )–目標(biāo) GPU ID。 默認(rèn)為當(dāng)前設(shè)備。
• non_blocking (bool )–如果True并且源位于固定內(nèi)存中，則副本將相對(duì)于主機(jī)是異步的。 否則，該參數(shù)無(wú)效。
• \ –為兼容起見(jiàn)，可以包含鍵async來(lái)代替non_blocking參數(shù)。

data_ptr()


device

double()

將此存儲(chǔ)空間轉(zhuǎn)換為雙精度類型

dtype

element_size()

fill_()

float()

將此存儲(chǔ)轉(zhuǎn)換為浮動(dòng)類型

static from_buffer()

static from_file(filename, shared=False, size=0) → Storage

如果共享的<cite>為<cite>為</cite>，則在所有進(jìn)程之間共享內(nèi)存。 所有更改都將寫入文件。 如果<cite>共享的</cite>為<cite>假</cite>，則存儲(chǔ)上的更改不會(huì)影響該文件。</cite>

<cite>大小</cite>是存儲(chǔ)中元素的數(shù)量。 如果<cite>共享的</cite>為<cite>假</cite>，則文件必須至少包含 <cite>size * sizeof(Type）</cite>個(gè)字節(jié) (<cite>Type</cite> 是存儲(chǔ)類型 )。 如果<cite>共享的</cite>為 <cite>True</cite> ，則將根據(jù)需要?jiǎng)?chuàng)建文件。

Parameters

• 文件名 (str )–要映射的文件名
• 共享的 (bool )–是否共享內(nèi)存
• 大小 (python：int )–存儲(chǔ)中的元素?cái)?shù)

half()

將此存儲(chǔ)空間轉(zhuǎn)換為一半類型

int()

將此存儲(chǔ)空間轉(zhuǎn)換為 int 類型

is_cuda = False

is_pinned()

is_shared()

is_sparse = False

long()

將此存儲(chǔ)空間轉(zhuǎn)換為長(zhǎng)型

new()

pin_memory()

將存儲(chǔ)復(fù)制到固定的內(nèi)存(如果尚未固定）。

resize_()

share_memory_()

將存儲(chǔ)移動(dòng)到共享內(nèi)存。

對(duì)于共享內(nèi)存中已存在的存儲(chǔ)和 CUDA 存儲(chǔ)(對(duì)于跨進(jìn)程共享無(wú)需移動(dòng)的 CUDA 存儲(chǔ)），此操作不起作用。 共享內(nèi)存中的存儲(chǔ)無(wú)法調(diào)整大小。

返回：自我

short()

將此存儲(chǔ)空間轉(zhuǎn)換為短型

size()

tolist()

返回包含此存儲(chǔ)元素的列表

type(dtype=None, non_blocking=False, **kwargs)

如果未提供 <cite>dtype</cite> ，則返回類型，否則將該對(duì)象強(qiáng)制轉(zhuǎn)換為指定的類型。

如果它已經(jīng)是正確的類型，則不執(zhí)行任何復(fù)制，并返回原始對(duì)象。

Parameters

• dtype (python：type 或 字符串）–所需類型
• non_blocking (bool )–如果True，并且源位于固定內(nèi)存中，而目標(biāo)位于 GPU 上，反之亦然，則相對(duì)于主機(jī)異步執(zhí)行復(fù)制。 否則，該參數(shù)無(wú)效。
• \ –為兼容起見(jiàn)，可以包含鍵async來(lái)代替non_blocking參數(shù)。 不推薦使用async arg。