pytorch系列文档之API:torch:Tensors详解

torch.is_tensor(obj)#是pytorch tensor时返回true
torch.is_storage(obj)#是pytorch 存储对象时返回true
torch.is_complex(input) -> (bool)#是复杂数据类型时返回true,例如,torch.complex64, and torch.complex128
torch.is_floating_point(input) -> (bool)#是否浮点数数据类型,torch.float64, torch.float32 and torch.float16等

torch.set_default_dtype(d)#设置默认数据类型为d,d指的是torch.float32、torch.float64等数据类型。
torch.get_default_dtype() → torch.dtype#返回当前默认数据类型

torch.set_default_tensor_type(t)#设置默认tensor类型为t,t指的是torch.FloatTensor、torch.DoubleTensor等
torch.numel(input)int#获得input中所有元素的个数,例如4x4的tensor中包含元素16个

torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None, sci_mode=None)
#打印设置,就是输出tensor时的设置
#precision 精度,默认小数点后4位
#threshold 触发显示摘要而不是整个时的值,默认1000
#edgeitems 显示摘要时,设置在摘要的每一维前面显示几个值,后面显示几个值,默认3
#linewidth用于插入换行符的每行字符数(默认= 80)。阈值矩阵将忽略此参数
#profile 配置文件,应该就是显示效果,可选default, short, full
#sci_mode 是否科学计数法,Enable (True) or disable (False) scientific notation
torch.set_flush_denormal(mode)bool#设置是否清洗cpu上的非正常浮点数

示例:
>>> torch.set_flush_denormal(True)
True
>>> torch.tensor([1e-323], dtype=torch.float64)
tensor([ 0.], dtype=torch.float64)
>>> torch.set_flush_denormal(False)
True
>>> torch.tensor([1e-323], dtype=torch.float64)
tensor(9.88131e-324 *
       [ 1.0000], dtype=torch.float64)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值