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

浙公网安备 33010602011771号