torch常用基础函数
torch
张量Tensors
torch.is_tensor(obj):
如果obj是一个pytorch张量,则返回True
torch.is_storage(obj):
如果obj是一个pytorch storage对象,则返回True
torch.numel(input):
返回input张量中的元素个数。
创建操作
torch.eye(n, m=None, out=None):
返回...
转载
2019-09-24 15:14:45 ·
689 阅读 ·
0 评论