Mug*_*ees 7
是torch.mode()
的内置函数(在此处阅读)可以处理您的两种情况。
torch.mode(alpha,0) #alpha being the name of tensor
更多推荐
张量,元素,方法,PyTorch
Mug*_*ees 7
是torch.mode()
的内置函数(在此处阅读)可以处理您的两种情况。
torch.mode(alpha,0) #alpha being the name of tensor
更多推荐
张量,元素,方法,PyTorch
发布评论