tensor&ndarray&int、float

(1)如果tensor只有一个元素,然后转换成int或者float类型的时候直接用int()或者float()就可以了;

(2)如果tensor含有多个元素,转换成ndarray时就要用x.detach().numpy()操作了(视情况使用,cpu());

猜你喜欢

转载自www.cnblogs.com/zf-blog/p/11365083.html
今日推荐