ww
def adjust_tensor_size(tensor): # 获取tensor的大小 size = list(tensor.size()) # 检查最后两个维度是否为奇数 for i in [-1, -2]: if size[i] % 2 != 0: size[i] -= 1 # 使用新的大小调整tensor tensor = tensor[:, :, :size[-2], :size[-1]] return tensor
def adjust_tensor_size(tensor): # 获取tensor的大小 size = list(tensor.size()) # 检查最后两个维度是否为奇数 for i in [-1, -2]: if size[i] % 2 != 0: size[i] -= 1 # 使用新的大小调整tensor tensor = tensor[:, :, :size[-2], :size[-1]] return tensor