选取部分张量元素

今天药忘吃喽~ 2022-04-10 20:46 176阅读 0赞

代码1：

t=torch.arange(12)
    print(t)
    print(t[3],t[0])
    print(t[-5])
    print(t[3:6])#个数为3
    print(t[:6])
    print(t[-5:])
    print(t[3:6:2])#2表示只有2个元素
    print(t[3::2])#以2位方差

运行结果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70][]

代码2 ：

t= torch.arange(24).reshape(2,3,4)#大小（2,3,4）
    print(t)
    print(t[1:,-2])
    print(t[1:,-2,2])
    print(t[0,:])

运行结果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70 1][]

代码3：

t= torch.arange(24).reshape(2,3,4)#大小（2,3,4）
    print(t)
    index=torch.tensor([2,2])#表示对那个维度进行处理
    print(t.index_select(1,index))

运行结果

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70 2][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70]: /images/20220411/486fdb909ba04832bc443a4f1b6e696b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70 1]: /images/20220411/00ef071aa0504bdbadffb68f4469d2d7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUyODA4OQ_size_16_color_FFFFFF_t_70 2]: /images/20220411/7b6b2e0f33414ab0b72afbf3e35f19fa.png

发表评论取消回复

表情：

评论列表（有 0 条评论，176人围观）

还没有评论，来说两句吧...

相关阅读

相关张量是啥？

参考：[【tensorflow】浅谈什么是张量tensor\_sowhat-CSDN博客][tensorflow_tensor_sowhat-CSDN] 所有的输入数据，如字

朱雀/ 2022年09月14日 11:07/ 0 赞/ 191 阅读

相关 Pytorch | Tensor张量

点击上方“机器学习与生成对抗网络”，关注星标获取有趣、好玩的前沿干货！来源：知乎—秦一地址：https://zhuanlan.zhihu.com/p/3993505

- 日理万妓/ 2022年09月09日 06:11/ 0 赞/ 246 阅读

相关张量tensorly笔记（1）——张量基础

![这里写图片描述][70] 一个非常形象的图片表示了从标量到张量的演变过程 -------------------- 使用python里面的numpy库和ten

素颜马尾好姑娘i/ 2022年05月22日 08:24/ 0 赞/ 555 阅读

相关张量分解-张量介绍

张量分解-张量介绍 2016.06.16 张量介绍张量(tensor)是一个多维的数据存储形式，数据的的维度被称为张量的阶。它可以看成是向量和矩阵在多维空间中的推

太过爱你忘了你带给我的痛/ 2022年05月22日 05:15/ 0 赞/ 865 阅读

相关 pytorch 张量

例程1: import torch t2=torch.tensor([[0,1,2],[3,4,5]]) print(t2) print('数

「爱情、让人受尽委屈。」/ 2022年04月11日 09:55/ 0 赞/ 249 阅读

相关选取部分张量元素

代码1： t=torch.arange(12) print(t) print(t[3],t[0]) print(t[-5]) prin

今天药忘吃喽~/ 2022年04月10日 20:46/ 0 赞/ 177 阅读

相关 Python TensorFlow，张量，张量的形状、类型、阶

张量是一个类型化的N维数组（tf.Tensor）,由三部分组成：名字，形状，数据类型。张量的阶： ![watermark_type_ZmFuZ3poZW5naGVpdGk

秒速五厘米/ 2022年03月15日 15:10/ 0 赞/ 263 阅读

相关 Python TensorFlow，张量的生成，张量运算的常用API，张量的类型转换 cast，张量的合并 concat

张量的数学函数API：[https://www.tensorflow.org/api\_docs/python/tf/math][https_www.tensorflow.or

超、凢脫俗/ 2022年03月14日 09:42/ 0 赞/ 282 阅读

相关张量求和

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5

╰+攻爆jí腚メ/ 2022年03月06日 04:44/ 0 赞/ 264 阅读

相关 TensorFlow基础：张量

一. 张量的定义不管是几阶，都是张量 ![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aH

迈不过友情╰/ 2022年02月20日 08:08/ 0 赞/ 265 阅读