import torch
import numpy as np
x=torch.rand(5,2)
print(x)
print(x[0],x[3])
a=np.array([[2,3],[3,4],[4,5],[6,0]])
a=torch.from_numpy(a) ####将numpy 转化为tensor
print(a,a.shape)
b=np.array([[2,3],[3,8],[4,5],[3,0]])
b=torch.from_numpy(b)
print(b,b.shape)
print(torch.equal(a,b))
for i in range(0,a.shape[0]):
print(a[i],b[i])
if (torch.equal(a[i],b[i])):
print("66666")
pytorch tensor 元素是否相等
猜你喜欢
转载自blog.csdn.net/weixin_42528089/article/details/103941327
今日推荐
周排行