pytorch和numpy操作对比

numpy pytorch
增加维度 np.expand_dims(input, axis) torch.unsqueezeinput, dim, out=None)
改变数据类型 input.astype(np.int64 / np.float64) input.type(torch.FloatTensor / torch.LongTensor)
posted @ 2019-03-20 23:40  rpSebastian  阅读(510)  评论(0)    收藏  举报