Questions tagged «tensor»

2
Keras输入说明:input_shape,units,batch_size,dim等
对于任何Keras层(Layer类),有人可以解释如何理解之间的区别input_shape,units,dim,等? 例如,文档说units指定图层的输出形状。 在神经网络的图像下面hidden layer1有4个单位。这是否直接转换为对象的units属性Layer?还是units在Keras中等于隐藏层中每个权重的形状乘以单位数? 简而言之,如何利用下面的图像来理解/可视化模型的属性,尤其是图层的属性?

21
如何在TensorFlow中打印Tensor对象的值?
我一直在使用TensorFlow中矩阵乘法的入门示例。 matrix1 = tf.constant([[3., 3.]]) matrix2 = tf.constant([[2.],[2.]]) product = tf.matmul(matrix1, matrix2) 当我打印产品时,它会将其显示为Tensor对象: <tensorflow.python.framework.ops.Tensor object at 0x10470fcd0> 但是我怎么知道的价值product呢? 以下内容无济于事: print product Tensor("MatMul:0", shape=TensorShape([Dimension(1), Dimension(1)]), dtype=float32) 我知道图形可以继续运行Sessions,但是没有任何方法可以在Tensor不运行图形的情况下检查对象的输出session吗?

8
PyTorch中的“视图”方法如何工作?
我对方法感到困惑 view()对以下代码片段中。 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16*5*5, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 10) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) x = x.view(-1, 16*5*5) x = F.relu(self.fc1(x)) x …
205 python  memory  pytorch  torch  tensor 

5
在PyTorch中保存经过训练的模型的最佳方法?
我一直在寻找其他方法来在PyTorch中保存经过训练的模型。到目前为止,我发现了两种选择。 使用torch.save()保存模型,使用torch.load()加载模型。 model.state_dict()保存训练的模型,model.load_state_dict()加载保存的模型。 我碰到过这种讨论,其中建议方法2优于方法1。 我的问题是,为什么选择第二种方法呢?仅仅是因为torch.nn模块具有这两个功能,我们被鼓励使用它们吗?

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.