代码之家 › 专栏 › 技术社区 › Tom Hale

获取PyTorch张量的数据类型

pytorch

Tom Hale · 技术社区 · 7 年前

如何找出PyTorch张量中元素的类型?

1 回复 | 直到 7 年前

prosti 6 年前

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor

你得到的第一个 print(t.dtype) 如果 t 是你的张量,否则你用 t.type() 另外两个。

Tom Hale 7 年前

>>> import torch
>>> print(torch.Tensor(1).dtype)
torch.float32

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

1 年前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

1 年前

Anonymous · 如何为零维火炬张量赋值?

1 年前

JohnnyWang97 · getattr引起的有趣错误

1 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

1 年前

Stocavista · 无法在python中将float 64转换为float 32

2 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

2 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

2 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

2 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

2 年前