2024年7月5日
ww
摘要： def adjust_tensor_size(tensor): # 获取tensor的大小 size = list(tensor.size()) # 检查最后两个维度是否为奇数 for i in [-1, -2]: if size[i] % 2 != 0: size[i] -= 1 # 使用新的大小 阅读全文
posted @ 2024-07-05 10:48 helloWorldhelloWorld 阅读(46) 评论(0) 推荐(0)