【备忘录】tf.split torch.chunk

最新推荐文章于 2023-04-27 17:53:42 发布
最新推荐文章于 2023-04-27 17:53:42 发布
阅读量584 收藏
点赞数
分类专栏： 问题解决 文章标签： tensorflow pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/li_jiaoyang/article/details/115328998
版权
这篇备忘录介绍了Tensorflow的tf.split和PyTorch的torch.chunk函数，虽然它们作用相同，将张量沿特定维度拆分为多个部分，但返回类型有所不同，tf.split返回list，而torch.chunk返回tuple。
摘要由CSDN通过智能技术生成

他俩的作用是一样的。只不过tf返回list，torch返回tuple

import tensorflow as tf 
import numpy as np 
import torch

t = np.random.randint(1, 9, (2, 4))

s = tf.split(t, 2, axis=1)

a = torch
最低0.47元/天 解锁文章
今天NLP了吗
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
torch.split()方法
jacke121的专栏
06-19 1307
https://blog.csdn.net/dongjinkun/article/details/115375847
Pytorch学习(二十九) ------------torch.split与torch.roll结合进行的一种高端操作
Hungryof的专栏
02-18 1720
How to reverse a multi-hot encoding ? multihot_batch = torch.tensor([[0,1,0,1], [0,0,0,1], [0,0,1,1]]) (multihot_batch == torch.tensor(1)).nonzero() tensor([[0, 1], [0, 3], [1, 3], [2, 2], [2, 3]]) 希望得到 [[1, 3], [3], [2, 3]
tensorflow改写为pytorch的方法总结
lrt366的博客
07-17 1万+
替换方法： 1. matrix = tf.get_variable("Matrix", [output_size, input_size], dtype=input_.dtype) = matrix = Variable(torch.randn(output_size, input_size)) 2. self.D_l2_loss = tf.constant(0.0) = sel...
torch.split()与tensor.split()
biubiu713的博客
04-27 386
等价于torch.split(tensor, int/list, dim)将tensor切分为块，每个块都是原始tensor的一个视图。
tensorflow内存分配器剖析
jiangbo1017的专栏
11-24 1216
tensorflow基于Doug Lea的malloc实现了自己的一套内存分配器．（简易版本的dlmalloc）通过合并的方式减少内存碎片．前提是假设使用这个分配器的进程拥有很大的内存，并且所有的内存分配请求都是通过这个接口． 代码路径： core/common_runtime/bfc_allocator.h关键数据结构以及变量含义： Chunk：指向一块内存，prev/next指针组成了一个
Tensorflow中的内存分配
weixin_42721176的博客
06-07 1129
Tensorflow系统复杂，支持平台多，各类数据结构也多，所以设计一个统一的内存管理分配接口很重要。本文主要探讨tensorflow中的内存分配的相关机制，会重点研究其中实现的两种分配算法。 内存分配是系统中非常重要的一环，大家平常接触最多的就是malloc(new)和free(delete)。也就是分配和施放，在tensorflow中也是这样。Tensorflow提供了一个公共的接口类，Allocator类，该类提供了两个重要的方法： AllocateRaw分配内存，DeallocateRaw施
tensorflow 与 pytorch 之间改写
Beater博客的日常
03-31 1625
替换方法： 1. matrix = tf.get_variable("Matrix", [output_size, input_size], dtype=input_.dtype) = matrix = Variable(torch.randn(output_size, input_size)) 2. self.D_l2_loss = tf.constant(0.0) = self...
torch.chunk与torch.split的区别
rcj918的博客
04-28 1548
torch.chunk(tensor, chunks, dim=0) 将tensor 拆分成相应的组块， 最后一块会小一些如果不能整除的话。 a的size是（2，4），那么torch.chunk(a,2,0), 0表示以第一个维度，分成两块，第一个维度就是指（2，4）里面的2 torch.split(tensor, split_size_or_sections, dim=0) 将tensor 拆分...
pytorch中torch.cat(),torch.chunk(),torch.split()函数的使用方法
RealCoder的博客
04-11 1672
一、torch.cat()函数 熟悉C字符串的同学们应该都用过strcat（）函数，这个函数在C/C++程序中用于连接2个C字符串。在pytorch中，同样有这样的函数，那就是torch.cat()函数. 先上源码定义：torch.cat(tensors,dim=0,out=None) 第一个参数tensors是你想要连接的若干个张量，按你所传入的顺序进行连接，注意每一个张量需要形状相同，或者更...
Tensorflow中的常用函数
Davidmvp的博客
07-22 175
tf.nn.embedding_lookup函数的用法 tf.nn.embedding_lookup函数的用法主要是选取一个张量里面索引对应的元素。tf.nn.embedding_lookup（tensor, id）:tensor就是输入张量，id就是张量对应的索引，其他的参数不介绍。 import tensorflow as tf; import numpy as np; c = np...
pytorch每日一学40(torch.split())将张量拆分为块(tensor的分割)
Fluid_ray的博客
12-08 1万+
第40个方法 torch.split(tensor, split_size_or_sections, dim=0) 此方法和前面的torch.chunk()一样，都是将tensor进行切割的方法，但是这两有什么区别呢？其实我们split方法功能更多样一些。 看一下此方法的参数： tensor：要被切割的tensor。 split_size_or_sections：当此参数为整数时，意思是将tensor按照每块大小为split_size_or_sections来切割，当此参数为列表时，将此tensor切成
TensorFlow笔记——tf.split()拆分tensor和tf.squeeze()
Lavi的专栏
07-24 2万+
tf.split() 参数说明： split( value, num_or_size_splits, axis=0, num=None, name='split' ) value： 输入的tensor num_or_size_splits: 如果是个整数n，就将输入的tensor分为n个子tensor。如果是个tensor T，就将输入的tenso...
torch.split()
热门推荐
sky的博客
01-07 6万+
torch.split(tensor,split_size_or_sections,dim=0) torch.split()作用将tensor分成块结构。 参数： tesnor：input，待分输入 split_size_or_sections：需要切分的大小(int or list ) dim：切分维度 output：切分后块结构 <class 'tuple'&g...
tf.split()函数的用法
mls0311的博客
08-25 6万+
在tensorflow 的代码里经常看到tf.split()这个函数，今天来扒一扒这个API的用法 tf.split( value, num_or_size_splits, axis=0, num=None, name='split' ) Splits a tensor into sub tensors. If num_or_size_split...
pytorch中张量的分块.chunk()方法和拆分.split()方法
qq_43061415的博客
02-28 7444
### 张量的分块和拆分方法 #### 1.分块：.chunk()方法 .chunk()方法能够按照某维度，对张量进行均匀切分，并且返回结果是原张量的视图。 **（1）.chunk()函数的使用，第一个参数：目标张量，第二个参数：等分的块数，第三个参数：按照的维度**
torch中的split函数学习
说文科技，做有态度的研究。
11-13 1427
tensor的分割函数——split
pytorch中torch.chunk()方法
weixin_30823833的博客
01-23 1万+
chunk方法可以对张量分块，返回一个张量列表： torch.chunk(tensor,chunks,dim=0)→ List of Tensors Splits a tensor into a specific number ofchunks. Lastchunkwill be smaller if the tensor size along the given dimensi...
torch.chunk()
Hello_word5的博客
12-19 826
将tensor按维度分块，返回一个元组 例 import torch a = torch.Tensor([[1,2,2],[4,5,6],[3,5,8], [4,6,7]]) b = torch.chunk(a,2,dim=0) print(b)
torch.split
最新发布
05-31
`torch.split` 是 PyTorch 中的一个函数，用于将一个张量沿着指定的维度分割成多个子张量。 函数定义： ```python torch.split(tensor, split_size_or_sections, dim=0) ``` 参数说明： - `tensor`：需要分割的张量。 - `split_size_or_sections`：可以是一个整数，表示每个子张量的大小；也可以是一个列表，表示以指定的大小分割张量。 - `dim`：指定分割的维度，默认为 0。 返回值： 返回一个元组，包含分割后的子张量。 示例： ```python import torch x = torch.randn(6, 3) print(x) # 沿着第一维分割成两个子张量 x1, x2 = torch.split(x, 3, dim=0) print(x1) print(x2) # 沿着第二维分割成三个子张量 x1, x2, x3 = torch.split(x, [1, 1, 1], dim=1) print(x1) print(x2) print(x3) ``` 输出： ``` tensor([[ 1.0139, -0.7239, -0.9594], [-0.3797, -0.0799, -0.1045], [ 0.6062, -0.6313, 0.2924], [-1.2285, -0.7122, -1.1791], [-0.5107, -1.1765, 0.8406], [-0.6899, -0.9661, 1.1744]]) tensor([[ 1.0139, -0.7239, -0.9594], [-0.3797, -0.0799, -0.1045], [ 0.6062, -0.6313, 0.2924]]) tensor([[-1.2285, -0.7122, -1.1791], [-0.5107, -1.1765, 0.8406], [-0.6899, -0.9661, 1.1744]]) tensor([[ 1.0139], [-0.3797], [ 0.6062], [-1.2285], [-0.5107], [-0.6899]]) tensor([[-0.7239], [-0.0799], [-0.6313], [-0.7122], [-1.1765], [-0.9661]]) tensor([[-0.9594], [-0.1045], [ 0.2924], [-1.1791], [ 0.8406], [ 1.1744]]) ```
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值