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

 

posted @ 2024-07-05 10:48  helloWorldhelloWorld  阅读(10)  评论(0编辑  收藏  举报