tensor常见的形式

# Tensor常见形式
# 0:scalar
# 1:vector
# 2:matrix
# 3:n-dimensional tensor

import torch
from torch import tensor

# scalar
x = tensor(42.)
x

tensor(42.)

x.dim()

0

x.item()

42.0

# vector 向量
v = tensor([1.5, -0.5, 3.0])
v

tensor([ 1.5000, -0.5000,  3.0000])

v.dim()

1

v.size()

torch.Size([3])

# Matrix
M = tensor([[1,2],[3,4]])
M

tensor([[1, 2],
        [3, 4]])

M.matmul(M)

tensor([[ 7, 10],
        [15, 22]])

tensor([1,0]).matmul(M)

tensor([1, 2])

M * M

tensor([[ 1,  4],
        [ 9, 16]])

tensor([1,2]).matmul(M)

tensor([ 7, 10])