Torch nn.MM 实例

Torch nn.MM 实例

​​a = torch.ones(2,3) th> b = torch.ones(3,4)th> nn.MM()({a,b})

3 3 3 3 3 3 3 3

th> a = torch.ones(1,2,3)th> b = torch.ones(1,3,4)th> nn.MM()({a,b})

(1,.,.) = 3 3 3 3 3 3 3 3

th> a = torch.ones(1,2) th> b = torch.ones(1,3)th> nn.MM(true,false)({a,b})

1 1 1 1 1 1

th> a = torch.ones(2,2)th> b = torch.ones(2,3)th> nn.MM(true,false)({a,b})

2 2 2 2 2 2

th> a = torch.ones(5,2)th> b = torch.ones(5,3)th> nn.MM(true,false)({a,b})

5 5 5 5 5 5

th> a = torch.ones(5,4,1)th> b = torch.ones(5,4,2)th> nn.MM(true,false)({a,b})

(1,.,.) = 4 4

(2,.,.) = 4 4

(3,.,.) = 4 4

(4,.,.) = 4 4

(5,.,.) = 4 4 [torch.DoubleTensor of size 5x1x2]

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。