gpt4 book ai didi

pytorch - 获取 PyTorch 张量的数据类型

转载 作者:行者123 更新时间:2023-12-01 23:20:40 25 4
gpt4 key购买 nike

我知道 PyTorch 张量是同质的,即每个元素都是相同的类型。

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

最佳答案

事情分为三种:

dtype                   || CPU tensor               || GPU tensor

torch.float32 torch.FloatTensor torch.cuda.FloatTensor

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

关于pytorch - 获取 PyTorch 张量的数据类型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53374499/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com