小编Sar*_*han的帖子

pytorch 中的 tf.cast 等价物?

我是 PyTorch 的新手。TensorFlow 有一个 API tf.cast()tf.shape()。该tf.cast在TensorFlow特定的目的,有什么等值的火炬?我有张量 x= tensor(shape(128,64,32,32)): tf.shape(x) create tensor of Dimension 1 x.shape create the true dimension。我需要在火炬中使用tf.shape(x)

tf.cast已经不仅仅是改变张着不同的角色D型的火炬。

有没有人在 torch/PyTorch 中有等效的 API?

python tensorflow pytorch

6
推荐指数
1
解决办法
7267
查看次数

可视化神经网络层激活

张量流或 keras 中的特征可视化很简单,可以在这里找到。https://machinelearningmastery.com/how-to-visualize-filters-and-feature-maps-in-convolutional-neural-networks/卷积神经网络可视化 - 权重或激活?

如何在 pytorch 中做到这一点?

我正在使用 PyTorch 和预训练的 resnet18 模型。我需要输入图像并激活特定层(例如 Layer2.0.conv2)。Layer2.0.conv2 在预训练模型中指定。

简单来说;如何将链接一代码转换为 PyTorch?如何获取 resnet18 PyTorch 中的特定层以及如何获取输入图像的激活。我在tensorflow中尝试过,它有效,但PyTorch无效。

tensorflow pytorch

3
推荐指数
1
解决办法
4494
查看次数

标签 统计

pytorch ×2

tensorflow ×2

python ×1