小编Vik*_*ain的帖子

pytorch中当输入参数超过两个时如何使用forward()方法

有人可以告诉我方法中多个参数背后的概念吗forward()?一般来说,方法的实现forward()有两个参数

  1. 自己
  2. 输入

如果前向方法的参数多于这些参数,PyTorch 如何使用前向方法。

让我们考虑这个代码库: https://github.com/bamps53/kaggle-autonomous-driven2019/blob/master/models/centernet.py 这里在线 236 位作者使用了带有两个参数的前向方法:

  1. 中心
  2. 返回嵌入

我找不到一篇文章可以回答我关于第 254( return_embeddings:) 行和第 257( if centers is not None:) 行将在什么条件下执行的查询。据我所知,该方法由 nn 模块内部调用。有人可以在这上面放一些灯吗?

python neural-network deep-learning pytorch tensor

7
推荐指数
1
解决办法
1万
查看次数

标签 统计

deep-learning ×1

neural-network ×1

python ×1

pytorch ×1

tensor ×1