3 Star 0 Fork 0

HorizonMind/MSVQA

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
test.py 233 Bytes
一键复制 编辑 原始数据 按行查看 历史
QMMMS 提交于 2023-09-24 10:43 . qms
import torch
import numpy as np
a = list()
for i in range (10):
a.append(torch.randn(1, 3, 4).cuda())
# 将 a 转换成 numpy 数组
a = torch.tensor(np.array([item.cpu().detach().numpy() for item in a])).cuda()
print(a.shape)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/horizon-mind/msvqa.git
git@gitee.com:horizon-mind/msvqa.git
horizon-mind
msvqa
MSVQA
master

搜索帮助