发布时间:2023-02-27 18:00
其次自己补充一下,请看如下代码:
>>>import torch
>>>a = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
>>> print(a.size())
torch.Size([2, 2, 2])
>>> print(a)
tensor([[[1, 2],
[3, 4]],
[[5, 6],
[7, 8]]])
>>> b = a[:,:,1]
>>> print(b)
tensor([[2, 4],
[6, 8]])
>>> b = a[:,:,0:1]
>>> print(b)
tensor([[[1],
[3]],
[[5],
[7]]])
>>> b = a[:,1,:]
>>> print(b)
tensor([[3, 4],
[7, 8]])
如果a的维度是(2,2,2), 如果b=a[:, :, 1],此时b的维度是(2, 2);如果b=a[:, :, 0:1],此时b的维度是(2, 2, 1);可以看出来如果某一个维度是一个确切的一维数字,那么该维度就会消失。
【Pygame实战】你说神奇不神奇?吃豆人+切水果结合出一款你没玩过的新游戏!(附源码)
uniapptabbar的高度是多少_微信小程序自定义tabBar在uni-app的适配详解
公司测试部门来了个00后卷王之王,老油条感叹真干不过,但是...
行为识别 - Video Transformer Network
MindSpore报错"TypeError: parse() missing 1 required positional."
什么是Eureka?Eureka能干什么?Eureka怎么用?
MindSpore报错 `half_pixel_centers`=True only support in Ascend
WWW 2021最佳论文亚军:基于属性补全的异质图神经网络新架构