仅作为记录,大佬请跳过
直接上代码
import torch
'''生成一个与该tensor维度一样的全零tensor'''
a,b,c,d=img_HE_CLS.shape
tmp=torch.zeros(a,b,c,d)
'''判断两个tensor是否相等'''
torch.equal(tmp,img_HE_CLS)
即可。
仅作为记录,大佬请跳过
直接上代码
import torch
'''生成一个与该tensor维度一样的全零tensor'''
a,b,c,d=img_HE_CLS.shape
tmp=torch.zeros(a,b,c,d)
'''判断两个tensor是否相等'''
torch.equal(tmp,img_HE_CLS)
即可。