公共号:机器学习与python集中营
每周会专门从下面的操作中挑出一些常用的进行详细讲解。
一、关于Tensors的一些判断
torch.is_tensor(obj)
torch.is_storage(obj)
torch.set_default_dtype(d) #默认的type为torch.float32
torch.get_default_dtype() → torch.dtype #(→返回值,下同)
torch.set_default_tensor_type(t)
torch.numel(input) → int #返回tensor中所有的元素个数
torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None)
# 设置print的相关选项如下图:
torch.set_flush_denormal(mode) → bool
二、创建tensor的一些方法
# 注意:张量的随机创建会在下面的random sampling里面再说明。
torch.tensor(data, dtype=None, device=None, requires_grad=False) → Tensor
torch.sparse_coo_tensor(indices, values, size=None, dtype=None, device=None, requires_grad=False)→ Tensor
torch.as_tensor(data, dtype=None, device=None) → Tensor
torch.from_numpy(ndarray) → Tensor
torch.zeros(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)→ Tensor
torch.zeros_like(input, dtype=None, layout=None, device=None, requires_grad=False) → Tensor
torch.ones(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)→ Tensor
torch.ones_like(input, dtype=None, layout=None, device=None, requires_grad=False) → Tensor
torch.arange(start=0, end, step=1, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.range(start=0, end, step=1, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.linspace(start, end, steps=100, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.logspace(start, end, steps=100, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.eye(n, m=None, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)→ Tensor
torch.empty(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)→ Tensor
torch.empty_like(input, dtype=None, layout=None, device=None, requires_grad=False) → Tensor
torch.full(size, fill_value, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.full_like(input, fill_value, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
三、Indexing, Slicing, Joining, Mutating(转变/变化)
torch.cat(tensors, dim=0, out=None) → Tensor
torch.chunk(tensor, chunks, dim=0) → List of Tensors #在某一个维度将一个tensor分成几等份,chunks为int,即需要分成的份数
torch.gather(input, dim, index, out=None) → Tensor #Gathers values along an axis specified by dim.
torch.index_select(input, dim, index, out=None) → Tensor #