2020年10月20日
摘要：
import torch.nn as nn m = nn.Softmax(dim=0) input = torch.randn(2, 2, 3) print(input) print(m(input)) input： tensor([[[ 0.5450, -0.6264, 1.0446], [ 0. 阅读全文
posted @ 2020-10-20 21:10
Google-boy
阅读(1173)
评论(0)
推荐(1)
浙公网安备 33010602011771号