线性代数 -- pytorch

Posted 2022-03-30 Cong0ks

 

import torch

x = torch.tensor([3.0])
y = torch.tensor([2.0])

print(x + y, x * y, x/y, x**y)

 

 

 

import torch

x = torch.arange(4)
print(x)  # 可以将向量视为标量值组成的列表
print(x[3])  # 通过张量的索引来访问任一元素
print(len(x))  # 访问张量的长度
print(x.shape)  # 只有一个轴的张量，形状只有一个元素

 

 

 

 # 轴对称翻转示意图

import torch

A = torch.arange(20).reshape(5,4)  # 20个元素，5行4列
print(A)

# 矩阵转置(轴对称的翻转，参考示意图--个人理解感觉像是行转列，列转行)
print(A.T)

 

 

 

 

 

import torch
B = torch.tensor([[1, 2, 3], [2, 0, 4], [3, 4, 5]])
print(B)

print(B.T)

print(B == B.T)