如何得到RecursiveScriptModule的输入维度
时间: 2023-02-08 12:13:51 浏览: 159
获取输入法的高度
你可以使用`input_dim`属性来获取输入维度。例如:
```
model = RecursiveScriptModule()
input_dim = model.input_dim
```
你也可以在创建模型时指定输入维度:
```
model = RecursiveScriptModule(input_dim=10)
```
还可以使用`shape`属性来获取输入张量的形状。例如:
```
input = torch.randn(batch_size, input_dim)
print(input.shape) # 输出 (batch_size, input_dim)
```
阅读全文