PyTorch 学习笔记

一、基本操作

1.使用

import torch

2.Tensor Properties

example_tensor = torch.Tensor(
    [
     [[1, 2], [3, 4]], 
     [[5, 6], [7, 8]], 
     [[9, 0], [1, 2]]
    ]
)
example_tensor

output:
tensor([[[1., 2.],
         [3., 4.]],
        [[5., 6.],
         [7., 8.]],
        [[9., 0.],
         [1., 2.]]])

①Device

example_tensor.device

output:
device(type='cpu')

②Shape

example_tensor.shape

output:
torch.Size([3, 2, 2])

print("shape[0] =", example_tensor.shape[0])
print("size(1) =", example_tensor.size(1))

output:
shape[0] = 3
size(1) = 2

print("Rank =", len(example_tensor.shape))
print("Number of elements =", example_tensor.numel())

Rank = 3
Number of elements = 12

3.Indexing Tensors索引张量

example_tensor[1]

out:
tensor([[5., 6.],
        [7., 8.]])

example_tensor[1, 1, 0]

out:
tensor(7.)

#获取标量值
example_scalar = example_tensor[1, 1, 0]
example_scalar.item()

out:
7.0

#每个矩阵左上角元素
example_tensor[:, 0, 0]

out:
tensor([1., 5., 9.])

4.Initializing