Efficiently find the dot product of two lists of vectors stored as PyTorch tensors & preserve backprop

前端 未结 0 366
日久生厌
日久生厌 2020-12-17 04:17

Suppose I had tensors X and Y which are both (batch_size, d) dimensional. I would like to find the (batch_size x 1) tenso

相关标签:
回答
  • 消灭零回复
提交回复
热议问题