当前位置:网站首页>Tensorflow to pytorch notes; tf. gather_ Nd (x, y) to pytorch
Tensorflow to pytorch notes; tf. gather_ Nd (x, y) to pytorch
2022-07-03 02:43:00 【strawberry47】
Recorded will tensorflow turn pytorch when , Some commonly used function conversions :
Can't convert directly
tf.transpose(input,[1, 0, 2])
->input.permute([1, 0, 2])
It cannot be directly replaced bytorch.transpose
, Because you can't operate multidimensionaltf.expand_dims(input), axis=1)
->input.unsqueeze(1)
tf.concat([content1,content2], axis=1
->torch.cat((content1,content2), dim=1)
Remember to axis Switch to dimtf.tile(input, [2, 1])
->input.repeat([2, 1])
tf.range(10)
->torch.arange(0)
tf.reduce_sum(x, axis=1, keep_dims=True)
->torch.sum(x,dim=1,keepdim=True)
tf.clip_by_value(x, min, max)
->torch.clamp(x, min, max)
tf.multinomial(logits=a, num_samples=1)
->torch.multinomial(input=a, num_samples=1, replacement=False)
tf.equal(x, y)
->torch.eq(x, y)
tf.nn.embedding_lookup(W_fe, Feature_input + 1)
->torch.index_select(W_fe, 0, Feature_input + 1)
tf.one_hot()
->functional.one_hot()
tf.gather_nd(x,y) transformation
def gather_nd(self,params, indices):
''' 4D example params: tensor shaped [n_1, n_2, n_3, n_4] --> 4 dimensional indices: tensor shaped [m_1, m_2, m_3, m_4, 4] --> multidimensional list of 4D indices returns: tensor shaped [m_1, m_2, m_3, m_4] ND_example params: tensor shaped [n_1, ..., n_p] --> d-dimensional tensor indices: tensor shaped [m_1, ..., m_i, d] --> multidimensional list of d-dimensional indices returns: tensor shaped [m_1, ..., m_1] '''
out_shape = indices.shape[:-1]
indices = indices.unsqueeze(0).transpose(0, -1) # roll last axis to fring
ndim = indices.shape[0]
indices = indices.long()
idx = torch.zeros_like(indices[0], device=indices.device).long()
m = 1
for i in range(ndim)[::-1]:
idx += indices[i] * m
m *= params.size(i)
out = torch.take(params, idx)
return out.view(out_shape)
Can convert directly
tf.reshape()
->torch.reshape()
tf.log()
tf.squeeze
边栏推荐
- Add automatic model generation function to hade
- Gbase 8C function / stored procedure definition
- HTB-Devel
- 简单理解svg
- Summary of interview project technology stack
- GBase 8c系统表-pg_constraint
- 【翻译】后台项目加入了CNCF孵化器
- ASP. Net core 6 framework unveiling example demonstration [02]: application development based on routing, MVC and grpc
- oauth2.0鉴权,登录访问 “/oauth/token”,请求头Authorization(basicToken)如何取值???
- 5. File operation
猜你喜欢
MATLAB小技巧(24)RBF,GRNN,PNN-神经网络
A2L file parsing based on CAN bus (2)
定了,就选它
Matlab tips (24) RBF, GRNN, PNN neural network
[translation] the background project has joined the CNCF incubator
Tongda OA V12 process center
[shutter] setup of shutter development environment (supplement the latest information | the latest installation tutorial on August 25, 2021)
random shuffle注意
Build a private cloud disk cloudrev
The use of Flink CDC mongodb and the implementation of Flink SQL parsing complex nested JSON data in monggo
随机推荐
当lambda没有输入时,是何含义?
Monitoring and management of JVM
Gbase 8C system table PG_ cast
"Analysis of 43 cases of MATLAB neural network": Chapter 43 efficient programming skills of neural network -- Discussion Based on the characteristics of the new version of MATLAB r2012b
Xiaodi notes
错误Invalid bound statement (not found): com.ruoyi.stock.mapper.StockDetailMapper.xxxx解决
[translation] modern application load balancing with centralized control plane
[fluent] JSON model conversion (JSON serialization tool | JSON manual serialization | writing dart model classes according to JSON | online automatic conversion of dart classes according to JSON)
Choose it when you decide
Use cve-2021-43893 to delete files on the domain controller
Create + register sub apps_ Define routes, global routes and sub routes
Didi programmers are despised by relatives: an annual salary of 800000 is not as good as two teachers
oauth2.0鉴权,登录访问 “/oauth/token”,请求头Authorization(basicToken)如何取值???
Counter统计数量后,如何返回有序的key
Apple releases MacOS 11.6.4 update: mainly security fixes
sql server数据库添加 mdf数据库文件,遇到的报错
[fluent] listview list (map method description of list set | vertical list | horizontal list | code example)
The data in servlet is transferred to JSP page, and the problem cannot be displayed using El expression ${}
JS的装箱和拆箱
GBase 8c触发器(二)