pytorch中arange()函数用法

语法:torch.arange(start=0,end,step=1)

解释:开始默认为0,步长默认为1,可以不写;终值必须写

返回的个数:

 举例:

 


版权声明:本文为weixin_43498449原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。