pytorch h-sigmoid

import torch

x = torch.rand(30, 3, 416, 416).cuda()

start=time.time()
torch.sigmoid(x)

print(f'time,{time.time()-start}')

import torch

x = torch.rand(30, 3, 416, 416).cuda()
aaaaa=torch.nn.ReLU6()
start=time.time()
torch.sigmoid(x)
# x=aaaaa(x+3)/6

print(f'time,{time.time()-start}')

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/98042375