torch中全连接神经网络参数默认的数据类型
时间: 2023-08-01 14:08:12 浏览: 100
关于pytorch中全连接神经网络搭建两种模式详解
5星 · 资源好评率100%
在 PyTorch 中,全连接神经网络的参数默认数据类型为浮点型,即 `torch.float32`。这可以在创建全连接层时通过 `dtype` 参数来指定其他数据类型,如下面的示例代码所示:
```python
import torch.nn as nn
# 创建全连接层,输入维度为 100,输出维度为 10,数据类型为 torch.float16
fc = nn.Linear(100, 10, dtype=torch.float16)
```
如果没有指定数据类型,则默认为 `torch.float32`。
阅读全文