torch中x.size()函数改成keras函数
时间: 2024-03-05 18:55:11 浏览: 74
Pytorch模型权重转变为Keras对应的模型权重
在`keras`中,张量对象也没有`.size()`方法,通常使用`.shape`属性来获取张量的形状信息。因此,可以将原来的代码:
```python
m_batchsize, C, height, width, channel = x.size()
```
改为:
```python
m_batchsize, height, width, channel = x.shape.as_list()
```
其中,`x.shape`返回一个列表,可以使用`.as_list()`方法将其转换为列表形式,从而获取张量的形状信息。由于原来的代码中没有使用`C`这个维度,因此可以将其省略。
阅读全文