这个屌丝很懒，什么也没留下！

深度学习pytorch- tensor, dtype 和 shape_torch_tensor.dtype

作者：盐析白兔 | 2024-02-16 16:42:14

torch_tensor.dtype

import torch
# Number
t1 = torch.tensor(1.)
t1
1
2
3
4

tensor(1.)

t1.dtype
1

torch.float32

t1.shape
1

torch.Size([])

# Vector
t2 = torch.tensor([1., 2, 3, 4])
t2
1
2
3

tensor([1., 2., 3., 4.])

t2.dty

本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】