在查找MaxPool的stride参数的默认值的时候,发现了这么一段代码:
class _MaxPoolNd(Module):
def __init__(self, kernel_size, stride=None, padding=0, dilation=1,
return_indices=False, ceil_mode=False):
super(_MaxPoolNd, self).__init__()
self.stride = stride or kernel_size
stride的定义很有趣,是两个数字进行or运算。
在网上查阅后,发现是这样的:
当and与整数一起使用时,如果所有值都不为零,它将为您提供最后一个值;如果至少有一个值,则返回零。
1 and 2 and 3 # return 3
1 and 0 and 3 # return 0
对于or,python首先检查第一个值,直到某个时候它得到True,因此如果第一个值为True,则返回该值;如果第一个值为False,但第二个值为True,则返回第二个值。如果它们都是假的,则返回最后一个。
相当于False的值:"",0,False,0j, 0.0, [], {}, set()。
所以上段代码的意思是,如果stride给定为正整数,则为给定的值;如果给定为0,或者不填默认为None,则stride = kernal_size
本文探讨了在MaxPoolNd模块中,stride参数的默认值设定。当stride未指定或设为0时,其值将默认设置为kernel_size。介绍了Python中的and和or操作符在整数上下文中的特殊行为:and操作符返回最后一个非零值,而or操作符返回第一个真值。这段代码确保了stride的正确赋值逻辑。
861

被折叠的 条评论
为什么被折叠?



