torch维度获取数据详解

 

import torch
import numpy as np

# 80 10 5 3:4
# 80 3 11 11
import copy
a = torch.linspace(-4, -1, steps=4)


b = torch.linspace(1, 256, steps=256).view(4,4, 4,4)
d= torch.linspace(1, 256, steps=256).view(4,4, 4,4)

a=[0,1]
#取单个值，
a=[[0,0,0,1],[0,0,0,2]]#维度变了


d[np.asarray(a)]=0

c=[[0,1],[0,2]]
if a==c:
print("0000000000")

b[np.asarray(c)]=0
if int(torch.max(d-b))==0:
print("same")
# if b[np.asarray(a)]==b[np.asarray(c)]:
# print("11111111111")
# print(np.array(a))
#
# b[np.array(a)]=0