赞
踩
Tensor.new_tensor(data, *, dtype=None, device=None, requires_grad=False)
'''
1)data:输出的tensor具有与data相同的内容
2)dtype:指定输出tensor的dtype,默认为Tensor的dtype
3)device:指定输出tensor的device,默认为Tensor的device
'''
tensor = torch.ones((2,), dtype=torch.int8)
data = [[0, 1], [2, 3]]
tensor.new_tensor(data)
# tensor([[ 0, 1],
# [ 2, 3]], dtype=torch.int8)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。