摘要: 功能: 返回一个新的tensor,新tensor增加了一个纬度,新的纬度的大小是1。 参数 tensor: 在哪个tensor上增加纬度 dim:在哪个纬度之前增加一个纬度 示例: target1: 在第一个纬度之前增加一个纬度(原来是[4],增加以后变为[1, 4]) target2: 在第二个纬 阅读全文
posted @ 2021-11-02 22:07 jizhao 阅读(384) 评论(0) 推荐(0) 编辑
摘要: 相同点:都可以生成一个一维的张量 不同点:range生成的一维张量,包含end元素,而arange生成的一维张量不包含end元素 注意: range函数由于与Python内置的range函数语义不同,已经被废弃了。因此大家都应该使用arange函数! 阅读全文
posted @ 2021-11-02 21:58 jizhao 阅读(546) 评论(0) 推荐(0) 编辑