注册 登录
编程论坛 Python论坛

pytorch语法有不解的地方,求大佬帮助

smy1998 发布于 2022-05-27 09:47, 942 次点击
def forward(self, input: Tensor) -> Tensor:

这是反向传播函数,我想知道-> Tensor:是啥意思,是要转成tensor类型一定要这么写吗?
1 回复
#2
古1232022-05-28 09:31
-> 后面是返回值的注释,意思是返回类型为Tensor
1