随笔分类 -  pytorch

pytorch和numpy操作对比
摘要：| | numpy | pytorch | | | | | | 增加维度 | np.expand_dims(input, axis) | torch.unsqueezeinput, dim, out=None) | | 改变数据类型 | input.astype(np.int64 / np.floa 阅读全文
posted @ 2019-03-20 23:40 rpSebastian 阅读(510) 评论(0) 推荐(0)