pytorch的matmul怎么广播

标签：pre   span   4th   怎么   atm   pytorch   class   adc   can   

https://pytorch.org/docs/stable/notes/broadcasting.html#broadcasting-semantics

https://pytorch.org/docs/stable/torch.html?highlight=matmul#torch.matmul

# can line up trailing dimensions
>>> x=torch.empty(5,3,4,1)
>>> y=torch.empty(  3,1,1)
# x and y are broadcastable.
# 1st trailing dimension: both have size 1
# 2nd trailing dimension: y has size 1
# 3rd trailing dimension: x size == y size
# 4th trailing dimension: y dimension doesn‘t exist

pytorch的matmul怎么广播

标签：pre   span   4th   怎么   atm   pytorch   class   adc   can   

原文地址：https://www.cnblogs.com/Wanggcong/p/11001119.html