随笔分类 -  PyTorch

摘要:TORCH.TENSOR.SCATTER_ ###Tensor.scatter_(dim, index, src, reduce=None) → Tensor 把src里面的元素按照index和dim参数给出的条件,放置到目标tensor里面,在这里是self。下面为了讨论方便,目标tensor和s 阅读全文
posted @ 2021-11-05 22:04 jizhao 阅读(609) 评论(0) 推荐(0)
摘要:功能: 返回一个新的tensor,新tensor增加了一个纬度,新的纬度的大小是1。 参数 tensor: 在哪个tensor上增加纬度 dim:在哪个纬度之前增加一个纬度 示例: target1: 在第一个纬度之前增加一个纬度(原来是[4],增加以后变为[1, 4]) target2: 在第二个纬 阅读全文
posted @ 2021-11-02 22:07 jizhao 阅读(430) 评论(0) 推荐(0)
摘要:相同点:都可以生成一个一维的张量 不同点:range生成的一维张量,包含end元素,而arange生成的一维张量不包含end元素 注意: range函数由于与Python内置的range函数语义不同,已经被废弃了。因此大家都应该使用arange函数! 阅读全文
posted @ 2021-11-02 21:58 jizhao 阅读(640) 评论(0) 推荐(0)