2020年5月12日
选择函数index_select
摘要： 书中（pytorch入门实战）讲：index_select(input, dim, index)，指定维度dim上选取，未有示例。 查到相关资料后， import torch as t # 导入torch模块c = t.randn(3, 6) # 定义tensorprint(c)b = t.inde 阅读全文
posted @ 2020-05-12 11:22 珠江水手 阅读(1249) 评论(0) 推荐(1) 编辑