torch.from_numpy的其他参数
时间: 2023-04-05 19:00:39 浏览: 294
Pytorch学习之torch用法—-比较操作(Comparison Ops)
torch.from_numpy函数还有其他参数吗?
答:是的,torch.from_numpy函数还有其他参数,包括dtype和requires_grad。dtype参数用于指定返回的张量的数据类型,而requires_grad参数用于指定是否需要计算梯度。
阅读全文
Pytorch学习之torch用法—-比较操作(Comparison Ops)